'LinearRegression' 對象沒有屬性 'save'

Question

我正在嘗試使用以下代碼行保存線性模型，但由於“LinearRegression”對象沒有“保存”屬性，因此出現錯誤。

from sklearn.linear_model import LinearRegression

model = LinearRegression()

model.save('Linear_Model.h5')

如何解決這個問題？

Answer 1

與 keras/tensorflow 模型相反，model.save() 不是為 sklearn 模型構建的。 您可能希望像處理其他 python 對象一樣保存它們，如下所示：

1. 使用pickle保存模型

---

# save the model to disk
filename = 'finalized_model.sav'
pickle.dump(model, open(filename, 'wb'))
 
# some time later...
 
# load the model from disk
loaded_model = pickle.load(open(filename, 'rb'))
result = loaded_model.score(X_test, Y_test)

2. 使用 joblib 保存模型：

---

# save the model to disk
filename = 'finalized_model.sav'
joblib.dump(model, filename)
 
# some time later...
 
# load the model from disk
loaded_model = joblib.load(filename)
result = loaded_model.score(X_test, Y_test)

如果您有興趣， 本文有這些代碼和示例。 還要檢查可能已經回答了您正在尋找的問題的SO 問題的鏈接