保存XGBoost模型并输入数据进行预测
§ Code
import xgboost as xgb
# 定义XGBoost模型,并训练模型
model = xgb.XGBClassifier() # 初始化模型
model.fit(x_train, y_train) # 训练模型
# 保存XGBoost模型到文件中,文件名为xgboost.model
model.save_model('xgboost.model') # 保存模型到文件中
# 测试数据集进行预测,输入即可预测结果。
test_data=[[1,2,3,4]] # 输入测试数据集
y_pred = model.predict(test_data) # 预测结果
print("The prediction result of test data is: {}".format(y_pred))
创作工场
免责声明:
以上内容除特别注明外均来源于网友提问,创作工场回答,未经许可,严谨转载。
点击这里>>使用🔥专业版,更聪明、更完整、更原创!