基于sklearn乳腺癌数据集(sklearn.datasets.load_breast_cancer),编写代码训练一个逻辑回归模型,并给出accuracy、recall、precision、F1和auc等分类模型精度指标
时间: 2024-05-08 09:18:36 浏览: 200
人工智能导论 学生作品二_乳腺癌预测_逻辑回归分类.docx
5星 · 资源好评率100%
好的,根据您的要求,我会根据sklearn乳腺癌数据集训练一个逻辑回归模型,并给出准确率(accuracy)、召回率(recall)、精度(precision)、F1值和AUC等分类模型精度指标。下面是示例代码(Python),您可以在您的IDE中运行该代码:
```python
# 导入需要用到的库和数据集
from sklearn.datasets import load_breast_cancer
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score, recall_score, precision_score, f1_score, roc_auc_score
# 加载乳腺癌数据集
data = load_breast_cancer()
# 将数据集拆分为训练集和测试集,设置随机种子为1
X_train, X_test, y_train, y_test = train_test_split(data.data, data.target, test_size=0.3, random_state=1)
# 创建逻辑回归模型并拟合训练数据
lr_model = LogisticRegression()
lr_model.fit(X_train, y_train)
# 对测试集进行预测并计算各项精度指标
y_pred = lr_model.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
recall = recall_score(y_test, y_pred)
precision = precision_score(y_test, y_pred)
f1 = f1_score(y_test, y_pred)
auc = roc_auc_score(y_test, y_pred)
# 打印输出各项精度指标
print('准确率 (Accuracy): {:.4f}\n召回率 (Recall): {:.4f}\n精度 (Precision): {:.4f}\nF1值 (F1-score): {:.4f}\nAUC: {:.4f}'.format(accuracy, recall, precision, f1, auc))
```
该代码将数据集拆分为训练集和测试集,并使用逻辑回归模型来训练数据。然后,对测试集进行预测并计算各项精度指标,最终输出各项指标的值。您可以将此示例代码粘贴到您的编程环境中,并执行该代码来得到与您使用的乳腺癌数据集相关的逻辑回归模型的分类精度指标。
阅读全文