python:决策树 分类模型
2021/10/16 17:39:31
本文主要是介绍python:决策树 分类模型,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
决策树 分类模型 iris_dtree.py
import numpy as np from sklearn import datasets from sklearn.metrics import confusion_matrix,accuracy_score from sklearn.model_selection import train_test_split from sklearn import preprocessing # 加载鸢尾花数据集 iris_X,iris_y = datasets.load_iris(return_X_y=True) # 数据预处理:按列归一化 iris_X = preprocessing.scale(iris_X) # 切分数据集:测试集 30% iris_X_train,iris_X_test,iris_y_train,iris_y_test = train_test_split(iris_X,iris_y,test_size=0.3,random_state=0) # 决策树 分类模型 from sklearn import tree model = tree.DecisionTreeClassifier() # 模型训练 model.fit(iris_X_train,iris_y_train) # 模型预测 iris_y_pred = model.predict(iris_X_test) # 模型评估 # 混淆矩阵 print(confusion_matrix(iris_y_test,iris_y_pred)) print("准确率: %.2f" % accuracy_score(iris_y_test,iris_y_pred))
这篇关于python:决策树 分类模型的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-05-08有遇到过吗?同样的规则 Excel 中 比Python 结果大
- 2024-03-30开始python成长之路
- 2024-03-29python optparse
- 2024-03-29python map 函数
- 2024-03-20invalid format specifier python
- 2024-03-18pool.map python
- 2024-03-18threads in python
- 2024-03-14python Ai 应用开发基础训练,字符串,字典,文件
- 2024-03-13id3 algorithm python
- 2024-03-13sum array elements python