From 2bf2502c28007a20f88a5fec29a438f5f2f76845 Mon Sep 17 00:00:00 2001
From: Yaoming <18202121195@sina.cn>
Date: Sun, 11 Feb 2018 17:41:15 +0800
Subject: [PATCH] [remove]    local main function

---
 docs/evaluation.md              | 1 +
 models/gsgan/Gsgan.py           | 5 -----
 models/leakgan/Leakgan.py       | 4 ----
 models/maligan_basic/Maligan.py | 5 -----
 models/mle/Mle.py               | 4 +---
 models/pg_bleu/Pgbleu.py        | 4 +---
 models/rankgan/Rankgan.py       | 5 -----
 models/seqgan/Seqgan.py         | 5 -----
 models/textGan_MMD/Textgan.py   | 6 ------
 9 files changed, 3 insertions(+), 36 deletions(-)

diff --git a/docs/evaluation.md b/docs/evaluation.md
index 6f1836f..46f68e9 100644
--- a/docs/evaluation.md
+++ b/docs/evaluation.md
@@ -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 . 
diff --git a/models/gsgan/Gsgan.py b/models/gsgan/Gsgan.py
index 6ab62ad..51dda82 100644
--- a/models/gsgan/Gsgan.py
+++ b/models/gsgan/Gsgan.py
@@ -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')
\ No newline at end of file
diff --git a/models/leakgan/Leakgan.py b/models/leakgan/Leakgan.py
index 9358da2..8664cb4 100644
--- a/models/leakgan/Leakgan.py
+++ b/models/leakgan/Leakgan.py
@@ -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()
\ No newline at end of file
diff --git a/models/maligan_basic/Maligan.py b/models/maligan_basic/Maligan.py
index 691eaa6..4aa28de 100644
--- a/models/maligan_basic/Maligan.py
+++ b/models/maligan_basic/Maligan.py
@@ -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')
\ No newline at end of file
diff --git a/models/mle/Mle.py b/models/mle/Mle.py
index f2a4aba..cbbe947 100644
--- a/models/mle/Mle.py
+++ b/models/mle/Mle.py
@@ -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()
+
diff --git a/models/pg_bleu/Pgbleu.py b/models/pg_bleu/Pgbleu.py
index d5f5f03..9197947 100644
--- a/models/pg_bleu/Pgbleu.py
+++ b/models/pg_bleu/Pgbleu.py
@@ -133,6 +133,4 @@ def train_oracle(self):
                 self.evaluate()
 
 
-if __name__ == '__main__':
-    pgbleu = Pgbleu()
-    pgbleu.train_oracle()
+
diff --git a/models/rankgan/Rankgan.py b/models/rankgan/Rankgan.py
index 4b87b71..6dfd01c 100644
--- a/models/rankgan/Rankgan.py
+++ b/models/rankgan/Rankgan.py
@@ -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')
\ No newline at end of file
diff --git a/models/seqgan/Seqgan.py b/models/seqgan/Seqgan.py
index bbf5cc8..dbf5c02 100644
--- a/models/seqgan/Seqgan.py
+++ b/models/seqgan/Seqgan.py
@@ -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')
\ No newline at end of file
diff --git a/models/textGan_MMD/Textgan.py b/models/textGan_MMD/Textgan.py
index 37815c7..68adb8d 100644
--- a/models/textGan_MMD/Textgan.py
+++ b/models/textGan_MMD/Textgan.py
@@ -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')