dreamquark-ai / tabnet

PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf
https://dreamquark-ai.github.io/tabnet/
MIT License
2.65k stars 488 forks source link

Update tab_network.py #517

Open sciengineer opened 1 year ago

sciengineer commented 1 year ago

The mask tensor M needs to be transformed to realize the objective stated in the paper: "γ is a relaxation parameter – when γ = 1, a feature is enforced to be used only at one decision step".

IMPORTANT: Please do not create a Pull Request without creating an issue first. My solution to the issue https://github.com/dreamquark-ai/tabnet/issues/516. Any change needs to be discussed before proceeding. Failure to do so may result in the rejection of the pull request.

What kind of change does this PR introduce? a bugfix

Does this PR introduce a breaking change? No, a minor change.

What needs to be documented once your changes are merged?

Closing issues

Put closes #XXXX in your comment to auto-close the issue that your PR fixes (if such).