Преглед на файлове

Added convGAN-proximary to test list.

Kristian Schultz преди 4 години
родител
ревизия
223f431da9
променени са 1 файла, в които са добавени 2 реда и са изтрити 0 реда
  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))
              ]