老师,不知道是否可以提供一下源码呢?

来源:6-3 scikit-learn实现决策树

peigong

2017-05-25 10:52

老师,不知道是否可以提供一下源码呢?

写回答 关注

3回答

  • 途索
    2017-05-25 15:29:55

    其实最好自己敲一遍

  • CCCCCold丶kl
    2017-10-30 21:48:01
    def main():
        from sklearn.datasets import load_iris
        iris = load_iris()
        print(iris)
        print(len(iris["data"]))
        from sklearn.model_selection import train_test_split
        train_data,test_data,train_target,test_target=train_test_split(iris.data,iris.target,test_size=0.2,random_state=1)
    
        from sklearn import tree
        #决策树
        clf = tree.DecisionTreeClassifier(criterion="entropy")
        clf.fit(train_data,train_target)
        y_pred = clf.predict(test_data)
    
        #Verify
        from sklearn import metrics
        print(metrics.accuracy_score(y_true=test_target,y_pred=y_pred))
        print(metrics.confusion_matrix(y_true=test_target,y_pred=y_pred))
        with open("D:/test/test.xlsx","w") as f:
            tree.export_graphviz(clf,out_file=f)
    if __name__=="__main__":
        main()


  • luckyforever
    2017-05-25 12:24:03

    自己敲一遍

Python数据分析-基础技术篇

使用Python进行数据分析的基础模块简介

93614 学习 · 277 问题

查看课程

相似问题