def gen_with_multi_vae(target_class, target_num, data_name):
dataset = MyDataSet(data_name, target_class=target_class, encode=True)
module_features = (dataset.single_continuous_data_len, 30, 20, 16)
lr = 0.00088
batch_size = 100
trainer = Trainer(module_features=module_features, learning_rate=lr,
batch_size=batch_size,
dataset=dataset, output_data_label=target_class, output_data_size=target_num // 2)(100)
temp = trainer.output_data
print(temp[0].attr_list)
dataset = MyDataSet(temp, target_class=target_class, encode=True)
trainer = Trainer(module_features=module_features, learning_rate=lr,
batch_size=batch_size,
dataset=dataset, output_data_label=target_class, output_data_size=target_num)(100)
return [data.to_list(DataType.CONTINUOUS) for data in trainer.output_data]
def gen_with_vae(target_class, target_num, data_name):
learning_rate = 0.000921
dataset = MyDataSet(data_name, target_class=target_class, encode=True)
trainer = Trainer(module_features=(dataset.single_continuous_data_len, 30, 20, 16), learning_rate=learning_rate,
batch_size=64,
dataset=dataset, output_data_label=target_class, output_data_size=target_num)
trainer(80)
return trainer.output_data