混淆矩阵值错误,y_test不等于y_train



我一直试图在逻辑回归结果上运行混淆矩阵,我已经分裂了数据,所以y_test的大小为8,y_train的大小为67,我无法打印矩阵。我知道它们必须是相同的大小,但我找不到方法。错误信息:ValueError: Found input variables with inconsistent numbers of samples: [8, 67]

from sklearn.model_selection import train_test_split
x_train, x_test, y_train, y_test = train_test_split(df[["Age"]], df.Spending_Cat, test_size=0.1)
from sklearn.linear_model import LogisticRegression
model=LogisticRegression()
model.fit(x_train, y_train)
print(model.predict(x_test))
print(model.score(x_test, y_test))
from sklearn.metrics import confusion_matrix
print(confusion_matrix(y_test, y_train))

混淆矩阵以y_true(正确的目标值)和y_pred(分类器返回的估计目标)作为输入。在这种情况下,y_true和y_pred肯定具有相同的大小,但是您试图同时使用来自测试和来自训练的预测来运行它,这是没有意义的,因为它们具有不同的大小。

相关内容

  • 没有找到相关文章

最新更新