Explorar o código

Added CTAB-GAN to generator benchmark list.

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