sklearn中的决策树模块及流程

    技术2022-07-11  98

    sklearn 中的决策树

    一、模块sklearn.tree

    sklearn中的决策树都在“tree”这个模块之下,这个模块共包括五个类:

    tree.DecisionTreeClassifier分类树tree.DecisionTreeRegressor回归树tree.export_graphviz将生成的决策树导出为DOT格式,画图专用tree.ExtraTreeClassifier高随机版本的分类树tree.ExtraTreeRegressor高随机版本的回归树

    二、sklearn的基本建模流程

    在这个过程中,分类树对应的代码是:

    from sklearn import tree   # 需要导入的模块 clf = tree.DecisionTreeClassifier()   # 实例化 clf = clf.fit(x_train,y_train)   # 用训练集数据训练模型 result = clf.score(x_test,y_test)   # 导入测试集,从接口中调用需要的信息

    (根据菜菜的机器学习整理)

    Processed: 0.011, SQL: 9