sklearn -验证

Question

对于下面的模型，输出/重新创建验证集是如何进行的，以便我可以保存以供将来参考？

from sklearn.experimental import enable_hist_gradient_boosting
from sklearn.ensemble import HistGradientBoostingClassifier

model= HistGradientBoostingClassifier(max_iter= 500,
                                n_iter_no_change= 10,
                                verbose= 1,
                                scoring= 'roc_auc',
                                validation_fraction= 0.2,
                                early_stopping= True,
                                tol= 0,
                                learning_rate= 0.11,
                                max_depth= 5,
                                random_state= 123)

model.fit(X_train, Y_train)

Answer 1

我似乎在这篇文章(https://machinelearningmastery.com/save-load-machine-learning-models-python-scikit-learn/)中找到了解决方案。使用joblib.dump(model,'model.sav')对我来说是有效的。