【发布时间】:2019-09-10 22:45:30
【问题描述】:
我是机器学习的新手,我正在构建一个简单的线性回归模型。该模型的变量如下:
X_train = [3, 5, 3, 4, 8, 7, 1, 10, 3, 2, 6, 6, 4, 9, 2, 1, 7, 5, 4, 8]
X_test = [2, 10, 4, 4, 10, 9, 10, 4, 5, 8]
Y_train = [56642, 66029, 64445, 61111, 113812, 91738, 46205, 121872, 60150, 39891, 81363, 93940, 57189, 54445, 105582, 43525, 39343, 98273, 67938, 56957]
Y_test = [37731, 122391, 57081, 63218, 116969, 109431, 112635, 55794, 83088, 101302]
这是我的代码:
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
# Importing the dataset
dataset = pd.read_csv('Data.csv')
X = dataset.iloc[:, :-1].values
y = dataset.iloc[:, 1].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 = 1/3, random_state = 0)
#fitting simple linear regression on training sets
from sklearn.linear_model import LinearRegression
regressor = LinearRegression()
regressor.fit(X_train, y_train)
#predicting the test set results
y_pred = np.array(regressor.predict(X_test))
这是我绘制图表的代码:
#visualizing data for the test set
plt.scatter(X_test, y_test, color = 'red')
#Method 1: Using regressor.predict(X_train)
plt.plot(X_train,regressor.predict(X_train), color = 'green')
#Method 2: Using regressor.predict(X_test)
plt.plot(X_test, regressor.predict(X_test), color='yellow')
plt.title('Experience v/s Salary')
plt.xlabel('Experience')
plt.ylabel('Salary')
plt.show()
这是我的图表的图像:
为什么plt.plot(X_train,regressor.predict(X_train), color = 'green') 和plt.plot(X_test, regressor.predict(X_test), color='yellow') 得到相同的回归线,即使regressor.predict(X_train) 和regressor.predict(X_test) 得到不同的结果?
【问题讨论】:
-
lenofX_train和Y_train不匹配 -
您的样本数据格式不正确。
X_train有 19 个元素,Y_train有 20 个元素,X_train和X_test应该嵌套lists。此外,您应该包括您的地块的图像。 -
@Akhilesh 这是一个打字错误。我没有创建数据集,但下载了它。我只是使用
train_test_split将数据集划分为训练集和测试集 -
@gmds 为什么 X_train 和 X_test 应该是嵌套列表?我使用
train_test_split将数据分为训练集和测试集。此外,我无处可循的课程告诉我,这两个数组都需要嵌套才能使模型工作。因此,如果您能提供更多详细信息,将会很有帮助。
标签: python machine-learning linear-regression