Workflow action

det-lab · Jun 6, 2024 · 3743a68 · 3743a68
1 parent 300cf28
commit 3743a68
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/tests/tutorial.py b/tests/tutorial.py
@@ -19,7 +19,7 @@
     ori_data = real_data_loading(data_name, seq_len)
 elif data_name == "sine":
     # Set number of samples and its dimensions
-    no, dim = 10000, 5
+    no, dim = 100, 5
     ori_data = sine_data_generation(no, seq_len, dim)
 
 print(data_name + " dataset is ready.")
@@ -29,15 +29,13 @@
 parameters["module"] = "gru"
 parameters["hidden_dim"] = 24
 parameters["num_layer"] = 3
-parameters["iterations"] = 10000
+parameters["iterations"] = 100
 parameters["batch_size"] = 128
 
 
 # Run TimeGAN
 generated_data = timegan.train_timegan(ori_data, parameters)
-timed_training = timegan.train_timegan_timed(
-    ori_data, parameters, filename="timegan_save", seconds=60, phase=1, current_iter=0
-)
+# timed_training = timegan.train_timegan_timed(ori_data, parameters, filename="timegan_save", seconds=60, phase=1, current_iter=0)
 
 
 print("Finish Synthetic Data Generation")
diff --git a/timegan/data/__init__.py b/timegan/data/__init__.py