回归评估
就是计算均方误差
y_predict = estimator.predict(x_test)
error = mean_squared_error(y_test, y_predict)
print("正规方程均方误差为:",error)
总结
对比一下正规方程与梯度下降
梯度下降需要选择学习率,迭代求解,特征数量较大的时候可以使用
回归评估
就是计算均方误差
y_predict = estimator.predict(x_test)
error = mean_squared_error(y_test, y_predict)
print("正规方程均方误差为:",error)
总结
对比一下正规方程与梯度下降
梯度下降需要选择学习率,迭代求解,特征数量较大的时候可以使用