扫码加入数据分析学习群

机器学习模型 - 线性回归

可以通过两个步骤创建模型： -
1. 使用Training Data 训练模型
2. 使用Test Data 测试模型

训练模型

使用上面的代码创建的数据用于训练模型

# Sk-Learn contains the linear regression model

from sklearn.linear_model import LinearRegression

# Initialize the linear regression model

predictor = LinearRegression(n_jobs =-1)

# Fill the Model with the Data

predictor.fit(X = TRAIN_INPUT, y = TRAIN_OUTPUT)

训练模型

使用上面的代码创建的数据用于训练模型

# Sk-Learn contains the linear regression model

from sklearn.linear_model import LinearRegression

# Initialize the linear regression model

predictor = LinearRegression(n_jobs =-1)

# Fill the Model with the Data

predictor.fit(X = TRAIN_INPUT, y = TRAIN_OUTPUT)

测试数据

测试是手动完成的。可以使用一些随机数据进行测试，并测试模型是否为输入数据提供了正确的结果