|
@@ -267,6 +267,6 @@ generators = { "Repeater": lambda _data: Repeater()
|
|
|
, "CTAB-GAN": lambda _data: CtabGan()
|
|
, "CTAB-GAN": lambda _data: CtabGan()
|
|
|
, "convGAN-majority-5": lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5)
|
|
, "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=None)
|
|
, "convGAN-majority-full": lambda data: ConvGAN(data.data0.shape[1], neb=None)
|
|
|
- , "convGAN-proximary-5": lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5, withMajorhoodNbSearch=True)
|
|
|
|
|
- , "convGAN-proximary-full": lambda data: ConvGAN(data.data0.shape[1], neb=None, withMajorhoodNbSearch=True)
|
|
|
|
|
|
|
+ , "convGAN-proximity-5": lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5, withMajorhoodNbSearch=True)
|
|
|
|
|
+ , "convGAN-proximity-full": lambda data: ConvGAN(data.data0.shape[1], neb=None, withMajorhoodNbSearch=True)
|
|
|
}
|
|
}
|