美文网首页
终章:模型搭建与评估

终章:模型搭建与评估

作者: 小憨豆 | 来源:发表于2021-06-24 01:44 被阅读0次

经过了前面的数据清洗、重构以及可视化后,进入基于数据的建模,建立模型后为了判断模型的可靠性需要对模型进行评价。

以泰坦尼克号为例:

首先引用所需要用到的库,同时载入原始数据和清洗后的数据。

用sklearn中的函数对数据集进行建模。首先对数据集进行切割,分成训练集和测试集。

然后插入模型,这里选择的是逻辑回归模型和随机森林模型。以逻辑回归模型为例得到训练集和测试集的分数:

最后输出模型预测的结果:

接下来一步是对模型的评估。

通过交叉验证、混淆矩阵、ROC曲线等方式进行模型的评估,通过不断调整参数来确定模型的优劣。

相关文章

网友评论

      本文标题:终章:模型搭建与评估

      本文链接:https://www.haomeiwen.com/subject/oyaiyltx.html