标签：iris python 模型 train test import model sklearn 决策树

决策树 分类模型 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))

标签：iris,python,模型,train,test,import,model,sklearn,决策树
来源： https://blog.csdn.net/belldeep/article/details/120801467

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。