如何保存训练好的逻辑回归模型
时间: 2023-07-19 13:44:46 浏览: 296
keras 如何保存最佳的训练模型
保存训练好的逻辑回归模型可以使用Python的pickle或joblib库,以下是两种保存模型的方法:
1. 使用pickle库保存模型
```python
import pickle
from sklearn.linear_model import LogisticRegression
# 训练逻辑回归模型
model = LogisticRegression()
model.fit(X_train, y_train)
# 保存模型
with open('model.pkl', 'wb') as f:
pickle.dump(model, f)
```
其中,'model.pkl'是保存模型的文件名,'wb'表示以二进制方式写入文件。
2. 使用joblib库保存模型
```python
import joblib
from sklearn.linear_model import LogisticRegression
# 训练逻辑回归模型
model = LogisticRegression()
model.fit(X_train, y_train)
# 保存模型
joblib.dump(model, 'model.joblib')
```
其中,'model.joblib'是保存模型的文件名。
需要注意的是,使用pickle或joblib保存模型时,模型的预处理器(例如缩放器、独热编码器等)也需要一并保存,以便在使用模型进行预测时进行相同的预处理。
阅读全文