Image Upload

euisuk-chung · Dec 6, 2021 · 54ec1b6 · 54ec1b6
1 parent 649e7a7
commit 54ec1b6
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -13,12 +13,17 @@ For example :
 
 ```python
 python run_timegan.py
-
 ```
 
 2. VRAE를 이용한 시계열 생성
 
 ```python
 python run_vrae.py
+```
+
+# Model Explanation
+## TimeGAN
+<img src = 'link'>
 
-```
+## VRAE
+<img src = 'link'>
diff --git a/config_timegan.py b/config_timegan.py
@@ -77,15 +77,15 @@ def parser_setting(parser):
     # Model Arguments
     parser.add_argument(
         '--emb_epochs',
-        default=300,
+        default=1000,
         type=int)
     parser.add_argument(
         '--sup_epochs',
-        default=300,
+        default=1000,
         type=int)
     parser.add_argument(
         '--gan_epochs',
-        default=300,
+        default=1000,
         type=int)
     parser.add_argument(
         '--batch_size',

diff --git a/config_vrae.py b/config_vrae.py
@@ -86,7 +86,7 @@ def parser_setting(parser):
         type=int)
     parser.add_argument(
         '--n_epochs',
-        default=500,
+        default=1000,
         type=int)
     parser.add_argument(
         '--batch_size',

diff --git a/image/LSTM_VAE_architecture.png b/image/LSTM_VAE_architecture.png
diff --git a/image/TimeGAN_architecture.PNG b/image/TimeGAN_architecture.PNG
diff --git a/image/architecture.png b/image/architecture.png
diff --git a/image/train_test_image.png b/image/train_test_image.png