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

Question

我训练了一个模型并获得了不错的 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() 的东西？

标签： pythonmachine-learningxgboost

Answer 1

只是拟合新数据

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)