diff --git a/README.md b/README.md
index 35554ec..0ce6fbd 100644
--- 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
+
-```
\ No newline at end of file
+## VRAE
+
diff --git a/config_timegan.py b/config_timegan.py
index 3847622..7631742 100644
--- 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
index b459d9c..c462cbc 100644
--- 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
new file mode 100644
index 0000000..f17ae8c
Binary files /dev/null and b/image/LSTM_VAE_architecture.png differ
diff --git a/image/TimeGAN_architecture.PNG b/image/TimeGAN_architecture.PNG
new file mode 100644
index 0000000..28aa71a
Binary files /dev/null and b/image/TimeGAN_architecture.PNG differ
diff --git a/image/architecture.png b/image/architecture.png
deleted file mode 100644
index 407324c..0000000
Binary files a/image/architecture.png and /dev/null differ
diff --git a/image/train_test_image.png b/image/train_test_image.png
new file mode 100644
index 0000000..0b77c47
Binary files /dev/null and b/image/train_test_image.png differ