Explorar o código

Added convGAN-proximary to test list.

Kristian Schultz %!s(int64=4) %!d(string=hai) anos
pai
achega
223f431da9
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  1. 2 0
      library/analysis.py

+ 2 - 0
library/analysis.py

@@ -268,4 +268,6 @@ generators = [ ("ProWRAS",       lambda _data: ProWRAS())
              , ("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]))
+             , ("convGAN-proximary-5",       lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5, withMajorhoodNbSearch=True))
+             , ("convGAN-proxymary-full",  lambda data: ConvGAN(data.data0.shape[1], neb=data.data0.shape[1], gen=data.data0.shape[1], withMajorhoodNbSearch=True))
              ]