一、sklearn - 模型的保存和加载 - API
from sklearn.externals import joblib
- 保存
- joblib.dump(rf, ‘test.pkl’)
- 加载
- estimator = joblib.load(‘test.pkl’)
二、示例助解
-
保存
lr = LinearRegression()
lr.fit(x_train, y_train)
joblib.dump(lr, "test.pkl")
-
加载
model = joblib.load("test.pkl")
model.predict(x_test)