Digix2019华为算法精英挑战赛代码

Digix2019华为算法精英挑战赛代码

最终成绩: 决赛第九

问题

根据手机型号，颜色，用户偏好，手机APP等信息预测用户年龄．

https://developer.huawei.com/consumer/cn/activity/devStarAI/algo/competition.html#/preliminary/info/digix-trail-02/introduction

解决方案

这次比赛我们主要选择神经网络和lightgbm两种模型

神经网络

我们主要搭建了两种神经网络，一种是全连接神经网络，另外一种是密集连接神经网络(Dense connected block+ Fully connected block)，我们在神经网络的搭建中，结合回归和分类两方面的损失，对模型进行优化。

全连接神经网络

神经网络主要使用全连接, 使用了两种结构， 第一种是将数据整合为一个数据表， 输入一个几层全连接网络，使用了Dropout, 结构、参数和使用的数据如表1-1，详细模型参数和模型搭建，见程序(main.py文件)

model = Sequential()
model.add(Dense(1024, activation='relu', input_shape=(two_train.shape[1],)))
model.add(Dropout(0.6))
model.add(Dense(512, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(256, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(256, activation='relu'))
model.add(Dropout(0.5))
model