【学习笔记】# 实例化决策树API clf = DecisionTreeClassifier() # 训练模型 ...

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

# 实例化决策树API
clf = DecisionTreeClassifier()
# 训练模型
clf.fit(X_train , y_train)
# 用建立好的模型, 去查看在训练集上的分数
clf.score(X_train , y_train)
# 模型在测试数据集上面的准确率
clf.score(X_test , y_test)
# 查看模型建立后, 特征在建立模型中的重要请
clf.feature_importances_
import matplotlib.pyplot as plt
plt.style.use(\'seaborn\')
# 建立列表接收数据
# 接收测试集数据
score_test = []
# 接收训练集数据
score_train = []
# max_depth 是 1 - 10
for i in range(1,11):
    # 建模
    clf = DecisionTreeClassifier(max_depth= i , random_state=666 )
    # 训练
    clf.fit(X_train , y_train)
    # 计算训练集分数
    score_train.append(clf.score(X_train , y_train))
    # 计算测试集分数
    score_test.append(clf.score(X_test , y_test))
# 绘制训练集和测试集的分数变化曲线
plt.plot(range(1,11) , score_train , color= \'red\')
plt.plot(range(1,11) , score_test , color= \'green\')

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：classifier Decision CISI 学习笔记 Tree

兄弟阔以

赞

很棒