【发布时间】:2020-09-14 05:42:51
【问题描述】:
我正在尝试使用 sklearn 进行多元线性回归。
features_2 = ['chronic_disease_binary', 'outcome']
X = df.loc[:, features_2].values
Y = df.loc[:, ['age']].values
# X = pd.get_dummies(X,drop_first=True)
#
X_train_lm, X_test_lm, y_train_lm, y_test_lm = create_dataset_test(X, Y)
X_train_lm = X_train_lm.reshape((2596, -1))
lm = linear_model.LinearRegression()
model = lm.fit(X_train_lm, y_train_lm)
y_pred_lm = lm.predict(X_test_lm)
我在尝试 tp 对 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 2 is different from 1)
- 我的 X_train 有这个表格:
[[-0.77046461 1.29791815]
[-0.77046461 -0.77046461]
[-0.77046461 1.29791815]
...
[-0.77046461 -0.77046461]
[-0.77046461 1.29791815]
[-0.77046461 -0.77046461]]
- 而我的 y_train 是这样的:
[[59.]
[54.]
[40.]
...
[24.]
[33.]
[41.]]
- 我进行预测的数据具有以下形式:
[[-0.76666002]
[ 1.30435914]
[-0.76666002]
...
[-0.76666002]
[-0.76666002]
[-0.76666002]]
【问题讨论】:
-
X_test_lm.shape带给你什么? -
@AmiTavory 它给了我 (1300, 1)
-
看我的回答!正如我所说,您的尺寸不匹配。
标签: python scikit-learn regression data-science linear-regression