
GAIN: ICML 2018的生成对抗式插补网络(GAIN)代码库
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:GAIN是ICML 2018上提出的一种创新性的缺失数据处理方法,利用生成对抗网络进行高效的数据插补。本代码库为研究者提供了复现该模型所需资源。
“生成对抗式插补网络(GAIN)”的代码库作者是尹振成、詹姆斯·乔登以及米哈埃拉·范德沙尔。相关论文名为《GAIN:使用生成对抗网络进行数据插补》,由Jinsung Yoon,James Jordon和Mihaela van der Schaar在2018年国际机器学习会议上发表。
该代码库包含了一个实现于两个UCI数据集上的GAIN框架的插补功能。这两个数据集分别是UCI字母以及UCI垃圾邮件数据集。
要运行有关GAIN框架的培训与评估管道,只需执行命令`python3 -m main_letter_spam.py`即可。请注意,任何模型架构都可以作为生成器和鉴别器模型使用,例如多层感知机或CNN等。
输入参数包括:
- data_name:字母(letter)或者垃圾邮件(spam)
- miss_rate:缺少值的概率
- batch_size:批量大小
- hint_rate:提示率
- alpha:超参数
- 迭代次数
全部评论 (0)
还没有任何评论哟~


