sklearn(错误输入形状)ValueError

我是ML和sklearn领域的新手.我尝试在X_train [2500,800],Y_train [2500,8]的数据集上使用GaussianNB.

from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()
clf.fit(X, Y)

在运行程序时,它显示

ValueError: bad input shape (2500, 8).

如何将Y_train [2500,8]转换为Y_train [2500,1]？

解决方法:

OP使用的是一个热编码器,因此拟合功能不适用于数组@Ishant Mrinal建议使用此功能

Y_train = np.argmax(Y_train, axis=1)

这样您就可以将一种热门编码传递给fit函数.

标签：python,scikit-learn,machine-learning,naivebayes
来源： https://codeday.me/bug/20191011/1890496.html