scikit-learn实现简单的决策树
2024-09-15 08:01:39
#encoding=utf-8
import numpy as np
import pandas as pd def main():
#Pre-processing
from sklearn.datasets import load_iris
iris = load_iris()
print(iris)
print(len(iris["data"]))
# from sklearn.cross_validation import train_test_split
from sklearn.model_selection import train_test_split
train_data,test_data,train_traget,test_target=train_test_split(iris.data,iris.target,test_size=0.2,random_state=1) #Model
from sklearn import tree
clf = tree.DecisionTreeClassifier(criterion="entropy")
clf.fit(train_data,train_traget)
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("./python_source/tree.doc","w") as fw:
tree.export_graphviz(clf,out_file=fw) if __name__ == '__main__':
main()
最新文章
- 机器指令翻译成 JavaScript —— No.2 跳转处理
- HTML2
- Daily Scrum Meeting ——SecondDay
- JavaScript精要(系列)
- 字符编码详解及由来(UNICODE,UTF-8,GBK)[转帖]
- sql server 数据遍历插入表变量
- XA事务处理
- UI篇--Android中3种方法实现back键动作
- ExtJS中动态设置TextField的readOnly属性
- iptables原理详解以及功能说明
- jQuery DOM 元素方法 (十)
- eclipse导入项目之后有感叹号
- ArrayList add方法的实现之扩容
- Linux基础理论
- python- Socket &; Mysql 重要知识点
- request.getRequestDispatcher(";";).forward()中文乱码
- SqlDataAdapter概述
- Mongodb安全认证
- Oracle TNS Listener Remote Poisoning
- 840. Magic Squares In Grid (5月27日)