|
|
@@ -263,7 +263,8 @@ def runAllTestSets(dataSetList):
|
|
|
generators = [ ("Repeater", lambda _data: Repeater())
|
|
|
#, ("SpheredNoise", lambda _data: SpheredNoise())
|
|
|
, ("SimpleGAN", lambda data: SimpleGan(numOfFeatures=data.data0.shape[1]))
|
|
|
+ , ("ctGAN", lambda data: CtGAN(data.data0.shape[1]))
|
|
|
+ , ("CTAB-GAN", lambda _data: CtabGan())
|
|
|
, ("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]))
|
|
|
]
|