Closed tomtang110 closed 3 years ago
Sorry for the delayed response! I haven't had the chance to port the code over to PyTorch, and unfortunately, I don't have much time on my end to investigate this. Sorry! I'll leave this issue open in-case if other people want to chime in (contribution is always welcomed).
Hi
I reimplemented your code in pytorch, I found a strange phenomenon, the auc of income is just 0.56 and the auc of marital is 0.96. I checked the detail and don't know why it does not work?
Here is the model code: