|
|
@@ -206,7 +206,7 @@ testSets = [
|
|
|
generators = { "Repeater": lambda _data: Repeater()
|
|
|
, "ProWRAS": lambda _data: ProWRAS()
|
|
|
, "GAN": lambda data: SimpleGan(numOfFeatures=data.data0.shape[1])
|
|
|
- , "ctGAN": lambda data: CtGAN(data.data0.shape[1])
|
|
|
+ , "CTGAN": lambda data: CtGAN(data.data0.shape[1])
|
|
|
, "CTAB-GAN": lambda _data: CtabGan()
|
|
|
, "ConvGeN-majority-5": lambda data: ConvGeN(data.data0.shape[1], neb=5, gen=5)
|
|
|
, "ConvGeN-majority-full": lambda data: ConvGeN(data.data0.shape[1], neb=None)
|