Zero-Inflated-Time-Series-Generation/constants.py at main · ArdeleanRichard/Zero-Inflated-Time-Series-Generation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import os
import numpy as np
import torch
import warnings
warnings.filterwarnings('ignore')
import matplotlib
matplotlib.use('Agg')

np.random.seed(42)
torch.manual_seed(42)
if torch.cuda.is_available():
    torch.cuda.manual_seed(42)

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(f"Using device: {device}")

DATA_FOLDER      = "./data/"
OUT_FOLDER       = "./out_m5/"
SECONDS_PER_HOUR = 3600.0
MAX_HOURS        = 24.0   # hard physical upper bound

os.makedirs(OUT_FOLDER, exist_ok=True)


MAP_MODEL_NAMES = {
    "zits-gan": "ZITS-GAN",
    "zits-vae": "ZITS-VAE",
    "timegan": "TimeGAN",
    "transfusion": "TransFusion",
    "fide": "FIDE",
    "chronogan": "ChronoGAN",
    "tsgm_timegan": "TSGM-TimeGAN",
    "tsgm_vae": "TSGM-TimeVAE",
    "vae_dense": "VAE-Dense",
    "vae_conv": "VAE-Conv",
    "timeVAE": "TimeVAE",
}