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