Added Speed test.

library/analysis.py

@@ -8,6 +8,7 @@ from library.convGAN import ConvGAN
 
 import pickle
 import numpy as np
+import time
 import random
 from imblearn.datasets import fetch_datasets
 
@@ -153,6 +154,32 @@ def runExerciseForConvGAN(datasetName, resultList=None):
     exercise.saveResultsTo(f"data_result/{ganName}-{datasetName}.csv")
     if resultList is not None:
         resultList[datasetName] = avg
+
+
+def runSpeedTestForConvGan(datasetName, ganGenerator):
+    ganName = "convGAN"
+    print()
+    print()
+    print("///////////////////////////////////////////")
+    print(f"// Running speed test for {ganName} on {datasetName}")
+    print("///////////////////////////////////////////")
+    print()
+    d = []
+    t1 = time.time()
+    data = loadDataset(f"data_input/{datasetName}")
+    gan = ganGenerator(data.data0.shape[1])
+    random.seed(2021)
+    shuffler = genShuffler()
+    exercise = Exercise(shuffleFunction=shuffler, numOfShuffles=3, numOfSlices=3)
+    exercise.debug = (lambda _x: None)
+    t2 = time.time()
+    exercise.run(gan, data)
+    t3 = time.time()
+    d = (t3 - t1, t2 - t1, t3 - t2)
+    print(f"Total Time: {d[0]}")
+    print(f"Preparation Time: {d[1]}")
+    print(f"Test Time: {d[2]}")
+    return d
     
 testSets = [
     "folding_abalone_17_vs_7_8_9_10",