Python XGBoost 分类器无法“预测”：“TypeError：不支持数据类型”

Question

我有一个像这样的数据集：

print(X_test.dtypes)
metric1                    int64
rank                     float64
device_type                 int8
NA_estimate              float64

当我尝试对此数据集进行预测时，出现以下错误：

y_test_pred_xgb = clf_xgb.predict(xgb.DMatrix(X_test))
TypeError: Not supported type for data.<class 'xgboost.core.DMatrix'>

我进行了一些搜索，但只找到了object引起问题的变量数据类型的讨论。我的数据是否还有其他问题或者问题是其他原因？我查看了各种博客和 Kaggle 代码，但没有运气。

Answer 1

我遇到了同样的问题，并通过使用以下方法转换数据类型来解决它np.float32()：

model.predict(np.float32(X_test))