我训练了LGBMClassifier模型并以这种方式保存在文件中:
clf = lgb.LGBMClassifier( ... )
clf.fit(X_train,y_train,**fit_params)
clf.booster_.save_model("model1.txt")
##Predictions
y_pred = clf.predict_proba(X_data,num_iteration=clf.best_iteration_)[:,1]
现在,我要使用保存的模型进行另一个预测。但是如果我这样做:
## new predictions:
clf_fs = lgb.Booster(model_file='model1.txt')
y_pred2 = clf_fs.predict_proba(X_data2,num_iteration=clf_fs.best_iteration_)[:,1]
我有一个错误
AttributeError: 'Booster' object has no attribute 'predict_proba'
我知道cls_fs
是类Booster
的对象,而不是类LGBMClassifier
的对象,并且我可以使用clf_fs.predict()
,但是如何获取来自已保存的增强文件及其所有特定属性的LGBMClassifier
对象?