溫馨提示×

Scikit-learn中怎么加載保存的模型

小億
94
2024-05-10 18:21:53
欄目: 編程語言

要加載保存的模型,可以使用joblib模塊中的loaddump方法。下面是一個示例代碼:

from sklearn.externals import joblib

# 保存模型
model = ...  # 使用Scikit-learn訓練得到的模型
joblib.dump(model, 'model.pkl')

# 加載模型
loaded_model = joblib.load('model.pkl')

在上面的代碼中,首先使用joblib.dump保存訓練得到的模型到model.pkl文件中。然后使用joblib.load加載保存的模型,得到loaded_model對象,即可繼續(xù)使用該模型進行預測等操作。

0