machine-learning - 机器学习和线性回归预测误差

Question

我试图通过简单的回归来预测答案，但得到以下错误：

（'形状（1,151）和（603,603）未对齐：151（dim 1）！= 603（dim 0）'）

这是我的代码

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

# Importing the dataset`enter code here`
dataset = pd.read_csv('pure_cotton.csv')
X = dataset.iloc[:,7].values
y = dataset.iloc[:,10].values

# Splitting the dataset into the Training set and Test set
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 0)


#fitting simple_linear_reg to training set
from sklearn.linear_model import LinearRegression
regressor = LinearRegression()
regressor.fit([X_train], [y_train])

#predicting the test results
y_pred = regressor.predict([X_test])

标签： machine-learning

Answer 1

你应该做一些 chengements：

regressor.fit(np.expand_dims(X_train, 1), np.expand_dims(y_train, 1))

和

regressor.predict(np.expand_dims(X_test, 1))

为了避免维度问题，第一个维度，代表样本个数，第二个维度代表特征个数