去噪变压器自动编码器

时间:2021-03-03 17:19:06
【文件属性】:
文件名称:去噪变压器自动编码器
文件大小:181KB
文件格式:ZIP
更新时间:2021-03-03 17:19:06
Python 降噪变压器自动编码器 此回购保存了我对Kaggle竞赛解决方案中的降噪自动编码器部分。 我的大部分工作都花在了训练降噪自动编码器网络上,以捕获输入之间的关系,并将学习到的表示形式用于下游监督模型。 复制单个模型 获取代码并移至代码目录。 按照指定安装python软件包。 下载比赛数据并修改的路径。 在单个GPU机器上运行python train.py并等待大约20个小时。 通过岭回归,其交叉验证的RMSE得分应为0.8412。 关于网络 该网络是一个AutoEncoder网络,中间层是变压器样式的编码器块。 训练网络执行以下两项任务:1)预测数据损坏掩码,2)重建干净的输入。 可以从变压器编码器输出中提取功能以用于下游任务。 网络图如下: 观察与思考过程 获取一个好的DAE。 我首先用(linear->relu) x 3作为中间层的老式自动编码器进行了实验,所学习的表示形
【文件预览】:
Denoise-Transformer-AutoEncoder-main
----.gitignore(38B)
----requirements.txt(74B)
----assets()
--------ac2.png(15KB)
--------c2.png(31KB)
--------diagram.png(61KB)
--------ac1.png(15KB)
--------c1.png(31KB)
--------fig1.png(36KB)
----LICENSE(1KB)
----README.md(6KB)
----util.py(305B)
----model.py(5KB)
----data.py(1KB)
----train.py(3KB)

网友评论