学习曲线概念:

1. 模型性能 = f(训练集大小)
2. 学习曲线的目的是为了获得更优的训练集大小,用在建模之前

代码实现:

import numpy as np
import pandas as pd
import sklearn.preprocessing as sp
import sklearn.ensemble as se
import sklearn.model_selection as ms
import matplotlib.pyplot as plt

# 创建一个随机森林分类器模型
# max_depth最大深度(层数) n_estimators树的个数, random_state随机种子
model = se.RandomForestClassifier(max_depth=9, n_estimators=140, random_state=7)

# _,没用的参数
_, train_scores, test_scores = ms.learning_curve(model, train_x, train_y, train_sizes = np.arange(0.1, 1.0, 0.1), cv=5)
print(test_scores)

plt.grid(linestyle=':')
plt.plot(np.arange(0.1, 1.0, 0.1), test_scores.mean(axis=1))
plt.show()

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐