滑雪板(坏输入形状) ValueError

Question

我是新的世界的ML和滑雪。我尝试在dataset上使用GaussianNB和X_train[2500,800]，Y_train[2500,8]。

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使用的是一个热编码器，因此fit函数不能与数组@Ishant一起工作。

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

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