机器学习__模型评估, 保存与加载

模型评估

import sklearn.metrics as sm
# 平均绝对值误差
print(sm.mean_absolute_error(train_y, pre_y))
# 平均平方误差
print(sm.mean_squared_error(train_y, pre_y))
# 中位绝对值误差
print(sm.median_absolute_error(train_y, pre_y))
# R2得分([0,1],越高越好)
print(sm.r2_score(train_y, pre_y))

# 0.3069767441860466
# 0.10930232558139548
# 0.3023255813953485
# 0.9812838483593501

保存模型

import pickle
with open('linear_model', 'wb') as f:
    pickle.dump(model, f)

加载模型

with open('linear_model', 'rb') as f:
    model = pickle.load(f)