python - Xgboost-ValueError：请在python中将输入数据X重塑为二维矩阵

Question

我使用 XGBoost Algortihm 进行文本挖掘，但在下面的代码中我不断收到错误消息。

from xgboost import XGBClassifier
from numpy import *
xgb=XGBClassifier()
xgb.fit(df['doc_vector'],df.cat1)

我怎样才能解决这个问题？谢谢你。

以下对象的类型和打印

标签： pythonxgboost

Answer 1

实际上我从ValueError 得到了解决方案：将两个数组传递给函数时，矩阵必须是二维的 谢谢。

xgb=XGBClassifier()
xgb.fit(np.vstack(df['doc_vector']),df.cat1)
predictions = xgb.predict(np.vstack(val['doc_vector']))