基于逻辑回归实现乳腺癌预测(机械学习与大数据)

基于逻辑回归实现乳腺癌预测

将乳腺癌数据集拆分成训练集和测试集,搭建一个逻辑回归模型,对训练集进行训练,然后分别对训练集和测试集进行预测。输出以下结果:

该模型在训练集上的准确率,在测试集上的准确率、召回率和精确率。

源码

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 recall_score,precision_score,classification_report,accuracy_score

cancer = load_breast_cancer()
x_train,x_test,y_train,y_test = train_test_split(cancer.data,cancer.target,test_size=0.2)
model = LogisticRegression(max_iter=10000)
model.fit(x_train,y_train)
train_score = model.score(x_train,y_train)
test_score = model.score(x_test,y_test)

print("1 基于逻辑回归实现乳腺癌预测")
print("李思强  20201107148")
print("训练集")
print("准确率:",train_score)

y_pred = model.predict(x_test)
accuracy_score_value = accuracy_score(y_test,y_pred)
recall_score_value = recall_score(y_test,y_pred)
precision_score_value = precision_score(y_test,y_pred)

print("测试集")
print("准确率:",accuracy_score_value)
print("召回率:",recall_score_value)
print("精确率:",precision_score_value)

运行结果

相关推荐
一碗绿豆汤34 分钟前
机器学习第二阶段
人工智能·机器学习
黑岚樱梦3 小时前
代码随想录打卡day23:435.无重叠区间
算法
番石榴AI3 小时前
基于机器学习优化的主图选择方法(酒店,景点,餐厅等APP上的主图展示推荐)
图像处理·人工智能·python·机器学习
Kuo-Teng3 小时前
Leetcode438. 找到字符串中所有字母异位词
java·算法·leetcode
数据与后端架构提升之路4 小时前
构建一个可进化的自动驾驶数据管道:规则引擎与异常检测的集成
人工智能·机器学习·自动驾驶
gihigo19984 小时前
MATLAB使用遗传算法解决车间资源分配动态调度问题
算法·matlab
墨染点香4 小时前
LeetCode 刷题【138. 随机链表的复制】
算法·leetcode·链表
却道天凉_好个秋5 小时前
目标检测算法与原理(一):迁移学习
算法·目标检测·迁移学习
Bony-5 小时前
糖尿病预测多个机器学习维度预测
人工智能·机器学习