正在阅读《第三节 识别毒蘑菇》
第(1/5)页


Python的scikit-learn(简称sklearn)模块可以非常好地支持决策树分类,尽管它使用的是更加复杂的决策树算法,但依然是在上一节讲述的原理基础上的优化算法。本节将使用scikit-learn的决策树分类来识别“蘑菇数据”中的毒蘑菇。

首先介绍本案例使用的数据。原始数据来源于加州大学欧文分校用于机器学习的数据库(UCI数据库),本书使用的是经过必要处理的数据,此数据可从教材资源平台下载。

接下来在命令行界面安装scikit-learn。

pipinstall -uscikit-learn

安装scikit-learn之前需要确保模块numpy和scipy已经安装,否则可以使用pipinstall安装。

pipinstall -unumpy

pipinstall -uscipy

进入Python之后,使用如下命令调用scikit-learn的决策树方法。

fromsklearnimporttree

模块scikit-learn的决策树是基于DecisionTreeClassifier对象,它能够解决二分类问题(如蘑菇是否可食用),也可以解决多分类问题。在使用中,对输入训练数据的维度要求如下。

输入X:样本数量×特征属性数量;

输入Y:样本类别标签,与X要一一对应。

下面先用一个简单的例子来熟悉使用方法。对四个点X=[[0,0],[0,1],[1,0],[1,1]]使用决策树进行分类,四个点分别属于两个类别0和1,它们对应的类别标签是Y=[0,0,1,1]。这可以通过如下简单代码实现。

In[1]:fromsklearnimporttree

X=[[0,0],[0,1],[1,0],[1,1]]

Y=[0,0,1,1]

clf=()

clf=(X,Y)

clf命令用来构造决策树,通过clf.fit(X,Y)实现了决策树的构建,此时clf已经是能够进行分类的决策树了,可以用它来进行新数据的分类。例如,输出[[0.3,0],[0.8,1],[1.2,1]]的类别,代码如下。

In[2]:test=[[0.3,0],[0.8,1],[1.2,1]]

In[3]:(test)

Out[3]:array([0,1,1])

通过上述输出可以看到决策树给出的分类结果是[0.3,0]类别为0;[0.8,1]类别为1;[1.2,1]类别为1。读者可以在平面直角坐标系上画出这些点,看看分类是否合理。

读者还可以尝试下列与决策树相关的函数,看看它们具有什么功能。

In[4]:dir(clf)

Out[4]:

[……

'apply',

'class_weight',

'classes_',

'criterion',

'decision_path',

'feature_importances_',

'fit',

'fit_transform',

'get_params',

'max_depth',

'max_features',

'max_features_;,

'max_leaf_nodes',

相关小说

宠妻狂魔:爹地,妈咪是我的 青春 / 完结
宠妻狂魔:爹地,妈咪是我的
一笔旧梦
惨遭妹妹陷害,她和陌生男人有了一夜情,五年后回归,却被腹黑又软萌的小娃娃被缠上。介绍一下,这是我爸,我觉得你们非常般配。总裁一脸冷酷将结婚证丢到了她面前,你已经是我的合法妻子了。什么情况?捡娃娃还带买一送一的?
27.59万字 4 年前
至尊魔妃,废柴大小姐 青春 / 完结
至尊魔妃,废柴大小姐
玖阑夜大人
她,是濮氏家族的濮四爷,打架找场子样样齐活。 她,潘家的废柴小姐,寄人篱下,受尽欺凌,样样占尽。 当她变成了她,功法仙器在手,宝贝丹药齐活。 敢算计她? 阴谋诡计,勾心斗角,全是她当年玩剩下的! 她定要让那些欺负她的人百倍偿还! 他,冷面仙君,修为超群,九重天门仙君之首,天帝都得给上几分薄面。 世人皆道他无心,偏偏他却认准了她,宠她入骨。 为渡她成仙,甘愿散去半生修为……
68.91万字 3 年前
顾先生,离婚吧! 青春 / 完结
顾先生,离婚吧!
冬菇家的猫
五年前,她被自己的丈夫陷害失身,家破人亡;为了报仇,她一人流落异国他乡,尝尽辛酸;四年后,她华丽回归,当名扬全城的美女设计师,碰上昔日里的前夫,又会擦出什么样的火花?
67.01万字 3 年前
嫡女重生,朕的皇后总想逃 都市 / 完结
嫡女重生,朕的皇后总想逃
涅凰
一朝眼瞎,一生凄苦,重活一世的沈姝誓要擦亮眼,防火防盗防渣男!渣男贱女,断她手脚筋,杀她长子,害她长女,毁她幼女!呵,这一桩桩,她都要同他们清算!渣男妄想攀附权贵?那就让他身败名裂!贱女阴毒?那就撕了她的美人皮!一心只有仇恨的她却无意间招惹到了一个了...
98.25万字 9 天前
农门冲喜小娘子 玄幻 / 完结
农门冲喜小娘子
笑猫嫣然
农门冲喜小娘子,相亲对像是初恋?呆了! 一口茶水没顺下去,嗝屁了。 再睁眼,床上躺着病歪歪的相公,她成冲喜娘子了? 屋子是借的,米缸是空的,地里长满荒草!!这到底是个什么情况?!! 床,只有一张?!被子,只有一条?! 白天伺候相公吃喝,晚上伺...
140.12万字 9 天前
星铁:重生成猫被符玄偷听心声 穿越 / 完结
星铁:重生成猫被符玄偷听心声
小太阳mt
++++琉光穿越到了的世界,但是本体却是一只流浪猫,并且还觉醒了一个好感度系统。该系统的作用是,让靠近琉光的人,都会对琉光产生好感。于是,第一个遇到琉光的符玄,一眼就喜欢上了这只圆滚滚、可爱的小猫,并将其带回了太卜司。但之后,符玄便发现,自己居然可以听见这小猫心里所想?!琉光:哎嘿嘿!符玄老婆今天又用她香香的、滑滑
158.16万字 7 天前