search

mp2893 / medgan

Generative adversarial network for generating electronic health records.
BSD 3-Clause "New" or "Revised" License
270 stars 91 forks source link

Fixing an error when running step 2-3 with count variables #17

Closed sylvaincom closed 5 years ago

sylvaincom commented 5 years ago

After running step 2-3 (after the training, to generate the synthetic records) with count variables, all the values of my synthetic records were between 0 and 1, thus binary after rounding. It seems that the default value of medgan.py is --data_type="binary" so we need to add the argument --data_type="count" in the final step (if we have count variables instead of binary ones).

mp2893 commented 5 years ago

Thanks for making the change. There were a couple of questions regarding this and now everyone will be better informed :)

sylvaincom commented 5 years ago

You're welcome, I'm glad I could help :)