Reproducibility for census data

[mrsergazinov]

Hi! I was trying to reproduce the results from the paper on the Census data provided in the repo. Specifically, I ran: python main.py config/config_census.yaml. I noticed that the number of epochs differs from the paper (10 in the repo -> 200 in the paper). Even with the epochs adjusted, I still cannot reproduce the results, the best AUC I get is Education: 0.867, Income: 0.906, Marital: 0.9645482646651425.

Could you share the config file for your best (final) model? I think this is because of some of the hyperparam's.