python - 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: bad input shape (2500, 8).

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

Answer 1

OP 使用的是单热编码器，因此 fit 函数不适用于数组 @Ishant Mrinal 推荐这个

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

这将允许您将 one hot 编码传递给 fit 函数。