XGBoost | 读取模型后再训练

最新推荐文章于 2024-06-24 09:36:59 发布

ASKCOS

最新推荐文章于 2024-06-24 09:36:59 发布

阅读量1.9k

点赞数 1

文章标签：机器学习 python scikit-learn ML xgboost

本文链接：https://blog.csdn.net/weixin_43236007/article/details/102677643

版权

深度学习专栏收录该内容

10 篇文章 1 订阅

订阅专栏

model = XGBRegressor(max_depth=10,n_estimators=200,min_child_weight=1,colsample_bytree=0.8,
					subsample=0.8, seed=1, learning_rate=0.1,n_jobs=4)

model.fit(X_train, y_train],eval_metric='rmse',
			eval_set=[(X_train, y_train),(X_val,y_val)],
			verbose=True,
			early_stopping_rounds=20)

model.save_model('XGB.pkl')

XGB_model = XGBRegressor()
XGB_model.load_model('XGB.pkl')

print("model is loaded successfully")
train_pred = XGB_model.predict(X_train)
val_pred = XGB_model.predict(X_val)
test_pred = XGB_model.predict(X_test)