소스 검색

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

Kristian Schultz 3 년 전
부모
커밋
61d7950d7b
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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)
              }