使用sklearn创建高次多项式线性模型 ,在训练集上进行训练
degree = 10 # 线性模型最高次幂
lmodel = lin_model(degree)
lmodel.fit(X_train, y_train)
对训练数据进行预测,找出训练误差
yhat = lmodel.predict(X_train)
err_train = lmodel.mse(y_train, yhat)
预测测试数据,得出测试误差
yhat = lmodel.predict(X_test)
err_test = lmodel.mse(y_test, yhat)