首页 > 解决方案 > matmul:输入操作数 1 的核心维度 0 不匹配,具有 gufunc 签名 (n?,k),(k,m?)->(n?,m?)(大小 20 与 10 不同)

问题描述

我正在尝试制作线性回归模型。请帮忙

database = pd.read_csv('Salary_Data.csv')
x = database.iloc[:,0].values
y = database.iloc[:,1].values
x_train = x_train.reshape(1,-1)
y_train = y_train.reshape(1,-1)
x_test = x_test.reshape(1,-1)
y_test = y_test.reshape(1,-1)
from sklearn.linear_model import LinearRegression
regressor= LinearRegression()
regressor.fit(x_train, y_train)

#predicting the test set results
y_pred = regressor.predict(x_test)

错误是:

ValueError: matmul: Input operand 1 has a mismatch in its core dimension 0, with gufunc signature (n?,k),(k,m?)->(n?,m?) (size 20 is different from 10)

标签: pythonmachine-learningscikit-learn

解决方案


我认为在重塑之后,您的x_train(我认为的 10 个维度)的维度与x_test(20 个维度)不同,因此该matmul函数会出错。你应该检查一下


推荐阅读