Kaynağa Gözat

Added convGAN-proximary to test list.

Kristian Schultz 4 yıl önce
ebeveyn
işleme
223f431da9
1 değiştirilmiş dosya ile 2 ekleme ve 0 silme
  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))
              ]