执行最后一行后，我收到以下错误:ValueError:y 应该是一个 1d 数组，而是得到一个形状为 (4457， 2)

Question

y = pd.get_dummies(messages['label'])
y = y.iloc[:,1].values
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.20,random_state = 0)

from sklearn.naive_bayes import MultinomialNB
spam_detect_model = MultinomialNB().fit(X_train, y_train)
y_pred = spam_detect_model.predict(y_test)

<在此之后收到此错误 应该是一个=" _x0031_d=" 的数组=">

Answer 1

首先，直接在你的y变量下，这样做以将其转换为整数：

y = y.apply(lambda x: x.argmax(), axis=1).values

而且y_pred = spam_detect_model.predict(X_test)不是y_test.