Explorar o código

Added benchmark for convGAN with full neighbourhood.

Kristian Schultz %!s(int64=4) %!d(string=hai) anos
pai
achega
f09401e550
Modificáronse 1 ficheiros con 4 adicións e 0 borrados
  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)