利用sklearn进行分类2:良/恶性乳腺癌肿瘤预测(二)

本文为《Python机器学习及实践:从零通往Kaggle竞赛之路》一书学习笔记,欢迎与我交流数据挖掘、机器学习相关话题。

本文与前文利用sklearn进行分类:良/恶性乳腺癌肿瘤预测(一)的区别在于数据集的特征更加丰富,一共有8个特征,类别还是两类:良性/恶性(在此数据集中分别记为2和4)。

1.数据预处理

该数据集有部分缺失数据被记为''?'',训练之前先去除含缺失值的样本

# 创建特征列表
data.replace(to_replace='?',value=np.nan,inplace=True) #将?替换为标准缺失值
data.dropna(how='any',inplace=True) #只要有一个维度数据有缺失则丢弃数据

使用sklearn自带的预处理库将数据集按3:1划分训练集和测试集

from sklearn.cross_validation import train_test_split #按3:1划分训练集与测试集
X_train,X_test,y_train,y_test = train_test_split(data[column_names[1:10]],data[column_names[10]],test_size=0.25)
#第1列为编号 2-9列为特征 第10列为类别(一共2类:2,4)

由于不同特征之间数值差距较大,为保证预测结果不会被某些维度过大的特征而主导,先标准化数据集

from sklearn.preprocessing import StandardScaler #将数据标准化为正态分布
#标准化特征
ss = StandardScaler()
X_train = ss.fit_transform(X_train)
X_test = ss.fit_transform(X_test)


2.训练模型

我们假设数据特征和分类目标之间存在线性关系,这里使用2种线性分类器LogisticRegression与SGDClassifier分别训练并作比较。

LogisiticRegression:采用常规精确计算的方式计算参数,计算时间长模型性能略高
SGDclassifier: 采用梯度法迭代渐进估计参数,计算时间短模型性能略低

#创建模型并训练、预测
lr = LogisticRegression()
sgd = SGDClassifier()

lr.fit(X_train,y_train)
lr_y_predict = lr.predict(X_test) #lr分类结果

sgd.fit(X_train,y_train)
sgd_y_predict = lr.predict(X_test) #sgd分类结果


3.评价训练结果
两种分类器分类效果比较

可以看到,LogisiticRegression效果略好,但大数据集下为了追求训练速度,一般采用梯度法进行线性分类。

from sklearn.metrics import classification_report
print 'Accuracy of LR Classifier:',lr.score(X_test,y_test)
print classification_report(y_test,lr_y_predict)                          

print 'Accuracy of SGD Classifier:',sgd.score(X_test,y_test)
print classification_report(y_test,sgd_y_predict) 

评价指标有score(即Accuracy)正确率、Precision精确率、Recall召回率、F1指数
以下记正样本为positive ,负样本为negative


常见的评价指标

F1计算方式

由计算公式可知,Accuracy即分类正确的个数与样本总数的比值,Precision为正样本个正确分类的个数与正样本总数的比值,Recall为分类正确的负样本个数与分类正确的正样本个数+分类错误的负样本个数的比值,而F1则衡量了Precision与Recall的距离,越接近,F1值越高。
对于肿瘤分类的实际问题,显然False negative的结果最不能接收,即负样本(恶性)没有被正确识别。

附一张更清晰的图


四种可能的分类结果
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,684评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,143评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,214评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,788评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,796评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,665评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,027评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,679评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,346评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,664评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,766评论 1 331
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,412评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,015评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,974评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,073评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,501评论 2 343

推荐阅读更多精彩内容