TGAN icon indicating copy to clipboard operation
TGAN copied to clipboard

How to pass TGANModel hyper parameters as a dictionary

Open afshinmarani opened this issue 5 years ago • 0 comments

I am using TGAN to synthesize new data based on an exclusive dataset. I want to tune the TGAN model using a Bayesian approach by means of "hyperopt" library. For this purpose, I need run the TGAN model while the hyperparameters are passed as a dictionary such as below:

tgan = TGANModel(continuous_columns, params, max_epoch=4, steps_per_epoch= 100, batch_size=16, restore_session=False)

where params is a dictionary including some hyperparameters: params = {'learning_rate': 0.002, 'z_dim': 50} Is there anyway I can pass the hyperparameters using such a dictionary as an argument in the TGANModel function? Should I use JSON file for this purpose? If yes, could you please help me how?

afshinmarani avatar Apr 28 '20 19:04 afshinmarani