Sfoglia il codice sorgente

Added benchmark for convGAN with full neighbourhood.

Kristian Schultz 4 anni fa
parent
commit
f09401e550
1 ha cambiato i file con 4 aggiunte e 0 eliminazioni
  1. 4 0
      library/analysis.py

+ 4 - 0
library/analysis.py

@@ -199,6 +199,9 @@ def runExerciseForCtGAN(datasetName, resultList=None, debug=False):
 def runExerciseForConvGAN(datasetName, resultList=None, debug=False):
     runExercise(datasetName, resultList, "convGAN", lambda data: ConvGAN(data.data0.shape[1], debug=debug))
 
+def runExerciseForConvGANfull(datasetName, resultList=None, debug=False):
+    runExercise(datasetName, resultList, "convGAN-full", lambda data: ConvGAN(data.data0.shape[1], neb=data.data0.shape[1], gen=data.data0.shape[1], debug=debug))
+
 
 def runSpeedTestForConvGan(datasetName, ganGenerator):
     ganName = "convGAN"
@@ -253,3 +256,4 @@ def runAllTestSets(dataSetList):
         runExerciseForSpheredNoise(dataset)
         runExerciseForSimpleGAN(dataset)
         runExerciseForConvGAN(dataset)
+        runExerciseForConvGANfull(dataset)