赞
踩
目录
ID3 信息增益 最大的准则
C4.5 信息增益比最大的准则
kaggkle决策树案例:泰坦尼克号沉没是历史上最臭名昭着的沉船之一。1912年4月15日,在她的处女航中,泰坦尼克号在与冰山相撞后沉没,在2224名乘客和机组人员中造成1502人死亡。这场耸人听闻的悲剧震惊了国际社会,并为船舶制定了更好的安全规定。 造成海难失事的原因之一是乘客和机组人员没有足够的救生艇。尽管幸存下沉有一些运气因素,但有些人比其他人更容易生存,例如妇女,儿童和上流社会。 在这个案例中,我们要求您完成对哪些人可能存活的分析。特别是,我们要求您运用机器学习工具来预测哪些乘客幸免于悲剧。
数据来源:https://www.kaggle.com/c/titanic
我们提取到的数据集中的特征包括票的类别,是否存活,乘坐班次,年龄,登陆home.dest,房间,船和性别等。
经过观察数据得到:
- from sklearn.model_selection import train_test_split
- import pandas as pd
- import numpy as np
- from sklearn.feature_extraction import DictVectorizer#转换成one_hot代码
- from sklearn.tree import DecisionTreeClassifier,export_graphviz#决策树、导出决策树
具体的做法看字典特征提取DictVectorizer(特征工程之特征提取)
- ef decision():
-
- """
- 决策树对于泰坦尼克号进行预测生死
- :return:None
- """
- #获取数据
- titan=pd.read_csv("C:\\Users\\zoutong\\Desktop\\programming\data\\machine learning\\train.csv")
- #处理数据,找出特征值和目标值
- x=titan[["Pclass","Age","Sex"]]
- y=titan["Survived"]
-
- # 缺失值处理
- x["Age"].fillna(x["Age"].mean(), inplace=True)
- #分割数据集到训练集和测试集
- x_train, x_test, y_train, y_test = train_test_split(x,y, test_size=0.25)
- #进行处理(特征工程) 特征->类别->one_hot代码
- dict=DictVectorizer(sparse=False)#DictVectorizer(sparse=False)会返回一个one-hot编码矩阵
- x_train=dict.fit_transform(x_train.to_dict(orient="records"))#pandas DataFrame转换为词典列表,其中1个词典代表1行
- print(dict.get_feature_names())
- x_test = dict.transform(x_test.to_dict(orient="records"))
- print(x_train)
这个是转换成one_hot代码的数据
- #用决策树进行分类
- dec=DecisionTreeClassifier()
- dec.fit(x_train,y_train)
- #预测准确率
- print("预测的准确率:",dec.score(x_test,y_test))
预测的准确率: 0.8116591928251121
安装graphiviz教程:Win / Mac安装Graphviz - 哔哩哔哩
- #导出决策树的结构
- export_graphviz(dec,out_file="./decision tree.dot",feature_names=["年龄","pclass","男性","女性"])
Dot文档中的内容:
- digraph Tree {
- node [shape=box, fontname="helvetica"] ;
- edge [fontname="helvetica"] ;
- 0 [label="女性 <= 0.5\ngini = 0.476\nsamples = 668\nvalue = [407, 261]"] ;
- 1 [label="pclass <= 2.5\ngini = 0.389\nsamples = 242\nvalue = [64, 178]"] ;
- 0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ;
- 2 [label="年龄 <= 2.5\ngini = 0.103\nsamples = 128\nvalue = [7, 121]"] ;
- 1 -> 2 ;
- 3 [label="pclass <= 1.5\ngini = 0.5\nsamples = 2\nvalue = [1, 1]"] ;
- 2 -> 3 ;
- 4 [label="gini = 0.0\nsamples = 1\nvalue = [1, 0]"] ;
- 3 -> 4 ;
- 5 [label="gini = 0.0\nsamples = 1\nvalue = [0, 1]"] ;
- 3 -> 5 ;
- 6 [label="年龄 <= 37.0\ngini = 0.091\nsamples = 126\nvalue = [6, 120]"] ;
- 2 -> 6 ;
- 7 [label="pclass <= 1.5\ngini = 0.044\nsamples = 88\nvalue = [2, 86]"] ;
- 6 -> 7 ;
- 8 [label="gini = 0.0\nsamples = 44\nvalue = [0, 44]"] ;
- 7 -> 8 ;
- 9 [label="年龄 <= 26.5\ngini = 0.087\nsamples = 44\nvalue = [2, 42]"] ;
- 7 -> 9 ;
- 10 [label="年龄 <= 25.5\ngini = 0.165\nsamples = 22\nvalue = [2, 20]"] ;
- 9 -> 10 ;
- 11 [label="年龄 <= 23.0\ngini = 0.091\nsamples = 21\nvalue = [1, 20]"] ;
- 10 -> 11 ;
- 12 [label="gini = 0.0\nsamples = 14\nvalue = [0, 14]"] ;
- 11 -> 12 ;
- 13 [label="年龄 <= 24.5\ngini = 0.245\nsamples = 7\nvalue = [1, 6]"] ;
- 11 -> 13 ;
Dot转化为PNG:
集成学习通过建立几个模型组合的来解决单一预测问题。他的工作原理是生成多个分类器/模型,各自独立的学习和做出预测。这些预测最后结合成但预测,因此优于任何一个单分类的做出预测。
定义:在机器学习中,随机森林是一个包含多个决策树的分类器,并且其输出的类别是由个别树输出的类别的众数而定。比如4个决策树结果是T,1个决策树结果是F,那么随机森林结果是T.
N个样本,M个特征
单个树建立过程:
如果不进行随机抽样,每棵树的训练集都一样,那么最终训练出的树分类结果也是完全一样。
如果不是有放回的抽样,那么每棵树的训练样本都是不同,都是没有交集的,这样每棵树都是“有偏”的,都是绝对”片面的”,也就是说每棵树训练出来的都是有很大的差异的;而随机森林最后分类取决于多棵树(弱2分类器)的投票表决。
- rf=RandomForestClassifier()
- param={"n_estimators":[120,200,300,500,800,1200],"max_depth":[5,8,15,25,30]}
- #网格搜索与交叉验证
- gc=GridSearchCV(rf,param_grid=param,cv=2)
- gc.fit(x_train,y_train)
- print("准确率:",gc.score(x_test,y_test))
- print("查看选择的参数模型:",gc.best_params_)
准确率: 0.8071748878923767
查看选择的参数模型: {'max_depth': 5, 'n_estimators': 300}
缺点基本上很少,有以下两点
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。