Ver código fonte

Updated Statistics with old convGAN tests.

Kristian Schultz 3 anos atrás
pai
commit
84927be0e1
1 arquivos alterados com 3 adições e 1 exclusões
  1. 3 1
      library/analysis.py

+ 3 - 1
library/analysis.py

@@ -262,10 +262,12 @@ def runAllTestSets(dataSetList):
 
 generators = { "ProWRAS":                 lambda _data: ProWRAS()
              , "Repeater":                lambda _data: Repeater()
-             , "SpheredNoise":            lambda _data: SpheredNoise()
+             #, "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-old-5":      lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5)
+             , "convGAN-old-full":   lambda data: ConvGAN(data.data0.shape[1], neb=data.data0.shape[1], gen=data.data0.shape[1])
              , "convGAN-majority-5":      lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5)
              , "convGAN-majority-full":   lambda data: ConvGAN(data.data0.shape[1], neb=data.data0.shape[1], gen=data.data0.shape[1])
              , "convGAN-proximary-5":     lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5, withMajorhoodNbSearch=True)