决策树处理CSV文件数据并计算准确率,精确率,召回率和F1的代码
时间: 2024-02-22 13:55:07 浏览: 89
好的,以下是一个使用决策树处理CSV文件数据并计算准确率、精确率、召回率和F1的Python代码示例:
``` python
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score, confusion_matrix
# 读取CSV文件数据
data = pd.read_csv('data.csv')
# 数据预处理
# ...
# 划分训练集和测试集
train_data = data.iloc[:100, :]
test_data = data.iloc[100:, :]
# 构建决策树模型
model = DecisionTreeClassifier()
model.fit(train_data.drop(['label'], axis=1), train_data['label'])
# 测试模型
y_pred = model.predict(test_data.drop(['label'], axis=1))
y_true = test_data['label']
# 计算模型评估指标
acc = accuracy_score(y_true, y_pred)
precision = precision_score(y_true, y_pred)
recall = recall_score(y_true, y_pred)
f1 = f1_score(y_true, y_pred)
confusion = confusion_matrix(y_true, y_pred)
print('Accuracy: {:.2f}'.format(acc))
print('Precision: {:.2f}'.format(precision))
print('Recall: {:.2f}'.format(recall))
print('F1 Score: {:.2f}'.format(f1))
print('Confusion Matrix:')
print(confusion)
```
这段代码首先读取CSV文件数据,然后进行数据预处理。接下来,将数据集划分为训练集和测试集,使用训练集来构建决策树模型,并使用测试集来测试模型的性能。
在测试过程中,使用`accuracy_score`、`precision_score`、`recall_score`和`f1_score`来计算模型的准确率、精确率、召回率和F1值。同时,使用`confusion_matrix`来计算混淆矩阵。
最后,输出模型评估指标的结果。
阅读全文