ML之DT:基于DT算法对titanic(泰坦尼克号)数据集乘客数据集进行二分类(是否获救)预测


过时的绿茶
过时的绿茶 2022-09-19 15:33:41 52165
分类专栏: 人才

ML之DT:基于DT算法对泰titanic(泰坦尼克号)数据集进行二分类(是否获救)预测

目录

输出结果

设计思路

核心代码


输出结果


设计思路

核心代码

  1. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25, random_state = 33)
  2. vec = DictVectorizer(sparse=False)
  3. X_train = vec.fit_transform(X_train.to_dict(orient='record'))
  4. X_test = vec.transform(X_test.to_dict(orient='record'))
  5. dtc = DecisionTreeClassifier()
  6. dtc.fit(X_train, y_train)
  7. y_predict = dtc.predict(X_test)
  8. dtc.score(X_test, y_test))
  9. classification_report(y_predict, y_test, target_names = ['died', 'survived'])

网站声明:如果转载,请联系本站管理员。否则一切后果自行承担。

本文链接:https://www.xckfsq.com/news/show.html?id=3313
赞同 0
评论 0 条
过时的绿茶L0
粉丝 0 发表 7 + 关注 私信
上周热门
如何使用 StarRocks 管理和优化数据湖中的数据?  2672
【软件正版化】软件正版化工作要点  2637
统信UOS试玩黑神话:悟空  2532
信刻光盘安全隔离与信息交换系统  2216
镜舟科技与中启乘数科技达成战略合作,共筑数据服务新生态  1092
grub引导程序无法找到指定设备和分区  743
WPS City Talk · 校招西安站来了!  15
金山办公2024算法挑战赛 | 报名截止日期更新  15
看到某国的寻呼机炸了,就问你用某水果手机发抖不?  14
有在找工作的IT人吗?  13
本周热议
我的信创开放社区兼职赚钱历程 40
今天你签到了吗? 27
信创开放社区邀请他人注册的具体步骤如下 15
如何玩转信创开放社区—从小白进阶到专家 15
方德桌面操作系统 14
我有15积分有什么用? 13
用抖音玩法闯信创开放社区——用平台宣传企业产品服务 13
如何让你先人一步获得悬赏问题信息?(创作者必看) 12
2024中国信创产业发展大会暨中国信息科技创新与应用博览会 9
中央国家机关政府采购中心:应当将CPU、操作系统符合安全可靠测评要求纳入采购需求 8

加入交流群

请使用微信扫一扫!