前言
上次分享,我们利用pandas和sklearn完成了泰坦尼克号数据的预处理,包括下面的步骤,具体可看深度学习|泰坦尼克号生还数据处理
- 选取建模字段
- 缺失值处理
- 性别处理
- 登陆船舱的处理
- 划分数据集
- 数据的标准化
那这篇文章我们就构造神经网络来完成数据的建模和预测。
MLP建模
模型结构
模型结构为:
- 输入层,也就是9个神经元(对应9个字段)
- 隐藏层1,40个神经元
- 隐藏层2,30个神经元
- 输出层,1个神经元
建立模型
from keras.models import Sequential
from keras.layers import Dense,Dropout
model = Sequen