|
|
@@ -196,8 +196,8 @@ def runExerciseForCtGAN(datasetName, resultList=None, debug=False):
|
|
|
runExercise(datasetName, resultList, "ctGAN", lambda data: CtGAN(data.data0.shape[1], debug=debug))
|
|
|
|
|
|
|
|
|
-def runExerciseForConvGAN(datasetName, resultList=None, debug=False):
|
|
|
- runExercise(datasetName, resultList, "convGAN", lambda data: ConvGAN(data.data0.shape[1], debug=debug))
|
|
|
+def runExerciseForConvGAN(datasetName, resultList=None, neb=5, debug=False):
|
|
|
+ runExercise(datasetName, resultList, "convGAN", lambda data: ConvGAN(data.data0.shape[1], neb=neb, gen=neb, 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))
|
|
|
@@ -257,3 +257,13 @@ def runAllTestSets(dataSetList):
|
|
|
runExerciseForSimpleGAN(dataset)
|
|
|
runExerciseForConvGAN(dataset)
|
|
|
runExerciseForConvGANfull(dataset)
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+generators = [ ("Repeater", lambda _data: Repeater())
|
|
|
+ #, ("SpheredNoise", lambda _data: SpheredNoise())
|
|
|
+ , ("SimpleGAN", lambda data: SimpleGan(numOfFeatures=data.data0.shape[1]))
|
|
|
+ , ("convGAN", lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5))
|
|
|
+ , ("convGAN-full", lambda data: ConvGAN(data.data0.shape[1], neb=data.data0.shape[1], gen=data.data0.shape[1]))
|
|
|
+ , ("ctGAN", lambda data: CtGAN(data.data0.shape[1]))
|
|
|
+ ]
|