热线电话:13121318867

登录
2018-12-17 阅读量: 814
python区分训练集和测试集

机器学习模型 - 线性回归

可以通过两个步骤创建模型: -
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)

测试数据

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

0.0000
3
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子