sklearn（输入错误形状）ValueError

Question

我是ML和sklearn世界的新手。我尝试在X_train[2500,800]，Y_train[2500,8]的数据集上使用GaussianNB。

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

在运行程序时，它正在显示

  

ValueError：输入形状错误（2500,8）。

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

Answer 1

OP正在使用一个热编码器，因此拟合函数不适用于数组@Ishant Mrinal建议

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

这将允许您将一个热编码传递给fit函数。