Skip to content

Commit

Permalink
[remove]
Browse files Browse the repository at this point in the history
   local main function
  • Loading branch information
Yaoming95 committed Feb 11, 2018
1 parent 019b93d commit 2bf2502
Show file tree
Hide file tree
Showing 9 changed files with 3 additions and 36 deletions.
1 change: 1 addition & 0 deletions docs/evaluation.md
Original file line number Diff line number Diff line change
Expand Up @@ -152,6 +152,7 @@ is in a bathroom with a toilet and a sink .
a motorcycle .
is is flying very of a bathroom with a bathroom .
```

MLE:
```text
there are tiled hanging across a large the toilet .
Expand Down
5 changes: 0 additions & 5 deletions models/gsgan/Gsgan.py
Original file line number Diff line number Diff line change
Expand Up @@ -358,8 +358,3 @@ def get_real_test_file(dict=iw_dict):
for _ in range(15):
self.train_discriminator()

if __name__ == '__main__':
gan = Gsgan()
# gan.train_oracle()
gan.train_cfg()
# gan.train_real('/home/ymzhu/Desktop/GAN/apex-text-gen/data/shi.txt')
4 changes: 0 additions & 4 deletions models/leakgan/Leakgan.py
Original file line number Diff line number Diff line change
Expand Up @@ -494,7 +494,3 @@ def get_real_test_file(dict=iw_dict):
print('epoch:' + str(epoch) + '--' + str(epoch_))
self.train_discriminator()

if __name__ == '__main__':
gan = Leakgan()
gan.train_oracle()
# gan.train_cfg()
5 changes: 0 additions & 5 deletions models/maligan_basic/Maligan.py
Original file line number Diff line number Diff line change
Expand Up @@ -353,8 +353,3 @@ def get_real_test_file(dict=iw_dict):
for _ in range(15):
self.train_discriminator()

if __name__ == '__main__':
maligan = Maligan()
# maligan.train_oracle()
# maligan.train_cfg()
maligan.train_real('/home/ymzhu/Desktop/GAN/apex-text-gen/data/toy.txt')
4 changes: 1 addition & 3 deletions models/mle/Mle.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,6 +169,4 @@ def get_real_test_file(dict=iw_dict):
generate_samples(self.sess, self.generator, self.batch_size, self.generate_num, self.generator_file)


if __name__ == '__main__':
pgbleu = Mle()
pgbleu.train_oracle()

4 changes: 1 addition & 3 deletions models/pg_bleu/Pgbleu.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,6 +133,4 @@ def train_oracle(self):
self.evaluate()


if __name__ == '__main__':
pgbleu = Pgbleu()
pgbleu.train_oracle()

5 changes: 0 additions & 5 deletions models/rankgan/Rankgan.py
Original file line number Diff line number Diff line change
Expand Up @@ -359,8 +359,3 @@ def get_real_test_file(dict=iw_dict):
for _ in range(15):
self.train_discriminator()

if __name__ == '__main__':
rankgan = Rankgan()
# rankgan.train_oracle()
# rankgan.train_cfg()
rankgan.train_real('/home/ymzhu/Desktop/GAN/apex-text-gen/data/shi.txt')
5 changes: 0 additions & 5 deletions models/seqgan/Seqgan.py
Original file line number Diff line number Diff line change
Expand Up @@ -362,8 +362,3 @@ def get_real_test_file(dict=iw_dict):
self.train_discriminator()


if __name__ == '__main__':
seqgan = Seqgan()
seqgan.train_oracle()
seqgan.train_cfg()
# seqgan.train_real('/home/ymzhu/Desktop/GAN/apex-text-gen/data/toy.txt')
6 changes: 0 additions & 6 deletions models/textGan_MMD/Textgan.py
Original file line number Diff line number Diff line change
Expand Up @@ -392,9 +392,3 @@ def toint_list(x):
for _ in range(15):
self.train_discriminator()


if __name__ == '__main__':
textgan = TextganMmd()
textgan.train_oracle()
textgan.train_cfg()
textgan.train_real('/home/ymzhu/Desktop/GAN/apex-text-gen/data/shi.txt')

0 comments on commit 2bf2502

Please sign in to comment.