第23步 机器学习分类实战:决策树(DT)建模


前言

假装有


一、Python调参

(1)建模前的准备

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
dataset = pd.read_csv('X disease code fs.csv')
X = dataset.iloc[:,<
好的,以下是一个使用KNN、决策树、随机森林对葡萄酒质量数据进行建模的完整Python代码: 首先,需要导入必要的库: ```python import pandas as pd from sklearn.model_selection import train_test_split from sklearn.preprocessing import StandardScaler from sklearn.neighbors import KNeighborsClassifier from sklearn.tree import DecisionTreeClassifier from sklearn.ensemble import RandomForestClassifier from sklearn.metrics import accuracy_score ``` 接下来,读取数据并进行预处理: ```python # 读取数据集 data = pd.read_csv('winequality.csv') # 将数据集分为特征和目标 X = data.iloc[:, :-1].values y = data.iloc[:, -1].values # 将目标变量转换为二元变量 for i in range(len(y)): if y[i] < 6: y[i] = 0 else: y[i] = 1 # 将数据集分为训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0) # 特征缩放 sc = StandardScaler() X_train = sc.fit_transform(X_train) X_test = sc.transform(X_test) ``` 然后,使用KNN进行建模和预测: ```python # KNN分类器 knn = KNeighborsClassifier(n_neighbors=5) knn.fit(X_train, y_train) # 预测测试集结果 y_pred_knn = knn.predict(X_test) # 输出模型准确率 acc_knn = accuracy_score(y_test, y_pred_knn) print("KNN准确率:", acc_knn) ``` 接下来,使用决策树进行建模和预测: ```python # 决策树分类dt = DecisionTreeClassifier(criterion='entropy', random_state=0) dt.fit(X_train, y_train) # 预测测试集结果 y_pred_dt = dt.predict(X_test) # 输出模型准确率 acc_dt = accuracy_score(y_test, y_pred_dt) print("决策树准确率:", acc_dt) ``` 最后,使用随机森林进行建模和预测: ```python # 随机森林分类器 rf = RandomForestClassifier(n_estimators=10, criterion='entropy', random_state=0) rf.fit(X_train, y_train) # 预测测试集结果 y_pred_rf = rf.predict(X_test) # 输出模型准确率 acc_rf = accuracy_score(y_test, y_pred_rf) print("随机森林准确率:", acc_rf) ``` 完整代码如下: ```python import pandas as pd from sklearn.model_selection import train_test_split from sklearn.preprocessing import StandardScaler from sklearn.neighbors import KNeighborsClassifier from sklearn.tree import DecisionTreeClassifier from sklearn.ensemble import RandomForestClassifier from sklearn.metrics import accuracy_score # 读取数据集 data = pd.read_csv('winequality.csv') # 将数据集分为特征和目标 X = data.iloc[:, :-1].values y = data.iloc[:, -1].values # 将目标变量转换为二元变量 for i in range(len(y)): if y[i] < 6: y[i] = 0 else: y[i] = 1 # 将数据集分为训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0) # 特征缩放 sc = StandardScaler() X_train = sc.fit_transform(X_train) X_test = sc.transform(X_test) # KNN分类器 knn = KNeighborsClassifier(n_neighbors=5) knn.fit(X_train, y_train) # 预测测试集结果 y_pred_knn = knn.predict(X_test) # 输出模型准确率 acc_knn = accuracy_score(y_test, y_pred_knn) print("KNN准确率:", acc_knn) # 决策树分类dt = DecisionTreeClassifier(criterion='entropy', random_state=0) dt.fit(X_train, y_train) # 预测测试集结果 y_pred_dt = dt.predict(X_test) # 输出模型准确率 acc_dt = accuracy_score(y_test, y_pred_dt) print("决策树准确率:", acc_dt) # 随机森林分类器 rf = RandomForestClassifier(n_estimators=10, criterion='entropy', random_state=0) rf.fit(X_train, y_train) # 预测测试集结果 y_pred_rf = rf.predict(X_test) # 输出模型准确率 acc_rf = accuracy_score(y_test, y_pred_rf) print("随机森林准确率:", acc_rf) ```
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Jet4505

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值