Kaggle:均值绝对误差

最新推荐文章于 2021-01-12 08:32:39 发布

ShellDawn

最新推荐文章于 2021-01-12 08:32:39 发布

阅读量287

点赞数

分类专栏： machine learning 文章标签： kaggle sklearn

本文链接：https://blog.csdn.net/ShellDawn/article/details/79804686

版权

machine learning 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

from sklearn.tree import DecisionTreeRegressor
import pandas as pd
from sklearn.metrics import mean_absolute_error

file = '../input/train.csv'
data = pd.read_csv(file)
y = data.SalePrice
predictors = ['YearBuilt','LotArea']
x = data[predictors]
model = DecisionTreeRegressor()
model.fit(x,y)
y_hat = model.predict(x)
mean_absolute_error(y,y_hat)
#这里会直接输出误差，不需要额外print