Explorar o código

Renamed convGAN test case to be compareable with the old tests.

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

+ 2 - 2
library/analysis.py

@@ -266,8 +266,8 @@ generators = { "ProWRAS":                 lambda _data: ProWRAS()
              , "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])
+             , "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)
              , "convGAN-proxymary-full":  lambda data: ConvGAN(data.data0.shape[1], neb=data.data0.shape[1], gen=data.data0.shape[1], withMajorhoodNbSearch=True)
              }