文件名称：去噪变压器自动编码器
文件大小：181KB
文件格式：ZIP
更新时间：2024-03-23 11:05:46
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)