Ver código fonte

Added CTAB-GAN to generator benchmark list.

Kristian Schultz 4 anos atrás
pai
commit
fe83eb534b
1 arquivos alterados com 2 adições e 1 exclusões
  1. 2 1
      library/analysis.py

+ 2 - 1
library/analysis.py

@@ -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]))
              ]