繁体   English   中英

如何使用经过训练的 XGB 分类模型预测新数据行?

[英]How to predict on new data row using trained XGB classification model?

我训练了一个模型并获得了不错的 auc。 现在,我想预测全新的数据,但我不知道该怎么做。 有人可以帮忙吗?

# fit model no training data
model = XGBClassifier()
model.fit(X_train, y_train)

# make predictions for test data
y_pred = model.predict(X_test)
predictions = [round(value) for value in y_pred]

#evaluate predictions train vs test data
accuracy = accuracy_score(y_test, predictions)
print("Accuracy: %.2f%%" % (accuracy * 100.0))

现在,我有一个全新的数据,我想用这个模型评分。 我该怎么做? 有什么predict.proba()?

刚刚拟合新数据

NEW_DTA = pd.read_csv(data)
New_y_test = NEW_DTA.iloc[:,-1]
New_x_test = NEW_DTA.drop(colums='Target')
New_pred = model.predict(New_x_test)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM