|
|
@@ -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-old-5": lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5)
|
|
|
- , "convGAN-old-full": lambda data: ConvGAN(data.data0.shape[1], neb=data.data0.shape[1], gen=data.data0.shape[1])
|
|
|
+ # , "convGAN-old-5": lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5)
|
|
|
+ # , "convGAN-old-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=None)
|
|
|
, "convGAN-proximary-5": lambda data: ConvGAN(data.data0.shape[1], neb=5, gen=5, withMajorhoodNbSearch=True)
|
