/prepare4pc_down

毕设最怕坏电脑

Primary LanguagePython

###########################################
环境配置:
Tensorflow 1.2.1
Python 2.7
###########################################
介绍:
这是LeakGAN在模拟数据下的实验。
###########################################
文件说明:
LeakGANModel.py 生成器模型,包括Manager和Worker模块。
Discriminator.py 判别器模型,包括特征提取网络和分类器。
target_lstm20.py 长度为20的预言模型,用来制造模拟数据。
data_loader.py 灌数据用的函数们。
Main.py 把各模块拼起来,训练和生成,run起来。
###########################################
细节:
我们提供了这些示例代码,让大家可以复现在模拟数据下的实验。
在默认参数下运行长度为20的实验:输入“python Main.py”
在默认参数下运行长度为40的实验:输入“python Main.py --length=40”

实验有两个阶段。第一个阶段,使用预言模型提供的好数据和MLE进行监督学习。第二个阶段,用对抗训练来增强生成器。
当你运行代码的时候,预训练模型会存放在ckpts文件夹下。
如果你想复位判别器预训练模型,可以输入:“python Main.py --resD=True --model=leakgan_preD”
如果你想复位所有预训练模型,可以输入:“python Main.py --restore=True --model=leakgan_pre”

当你结束实验后,可以得到负对数似然的训练效果记录,输出在终端上,同时也存在save/experiment-log.txt里。