数据科学技术与应用
机器学习建模分析(一) 第六次作业 答案
题目来源:中国大学MOOC-东华大学-宋晖等-数据科学技术与应用
1 ( 100分 ) Energy Efficiency数据集( ENB2012_data.xlsx,ENB2012.names)记录不同房屋的制热能源消耗和制冷能源消耗。包括768条记录,8个特征属性,两个预测值。具体说明见ENB2012.names。
1)在全数据集上训练线性回归模型预测制热能耗,计算模型性能:RMSE以及R2;
2)将数据集划分训练集和测试集,在训练集上训练线性回归模型,分析模型在训练集和测试集上的性能。
import pandas
from sklearn.linear_model import LinearRegression
from sklearn import model_selection
from sklearn import metrics