diff --git a/model/README.md b/models/README.md similarity index 100% rename from model/README.md rename to models/README.md diff --git a/model/TimeGAN.py b/models/TimeGAN.py similarity index 100% rename from model/TimeGAN.py rename to models/TimeGAN.py diff --git a/model/__init__.py b/models/__init__.py similarity index 100% rename from model/__init__.py rename to models/__init__.py diff --git a/model/vrae.py b/models/vrae.py similarity index 100% rename from model/vrae.py rename to models/vrae.py diff --git a/run_timegan.py b/run_timegan.py index a435015..3f4c3a2 100644 --- a/run_timegan.py +++ b/run_timegan.py @@ -24,9 +24,9 @@ from tqdm.auto import trange from scipy.stats import norm -from model.TimeGAN import TimeGAN -from model.custom_dataset import * -from model.utils_timegan import * +from models.TimeGAN import TimeGAN +from utils.custom_dataset import * +from utils.utils_timegan import * args = config.get_config() # argument 호출 fix_seed(args.seed) # seed 고정 diff --git a/run_vrae.py b/run_vrae.py index 868ee17..b0f3fa2 100644 --- a/run_vrae.py +++ b/run_vrae.py @@ -24,9 +24,9 @@ from tqdm.auto import trange from scipy.stats import norm -from model.vrae import VRAE -from model.custom_dataset import * -from model.utils_vrae import * +from models.vrae import VRAE +from utils.custom_dataset import * +from utils.utils_vrae import * # argument 호출 args = config.get_config() diff --git a/model/custom_dataset.py b/utils/custom_dataset.py similarity index 100% rename from model/custom_dataset.py rename to utils/custom_dataset.py diff --git a/model/utils_timegan.py b/utils/utils_timegan.py similarity index 100% rename from model/utils_timegan.py rename to utils/utils_timegan.py diff --git a/model/utils_vrae.py b/utils/utils_vrae.py similarity index 100% rename from model/utils_vrae.py rename to utils/utils_vrae.py