search

dreamquark-ai / tabnet

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

Add tensorboard logging #442

Closed KAEYL98 closed 1 year ago

KAEYL98 commented 1 year ago

Feature request

What is the expected behavior? Add support for tensorboard log files as an optional option to model parameters

What is motivation or use case for adding/changing the behavior? Better tracking of different runs when tuning hyper parameters

How should this be implemented in your opinion? Utilize existing torch.utils.tensorboard methods provided by pytorch

Are you willing to work on this yourself? Not sure

Optimox commented 1 year ago

I think this would create an unnecessary dependency to the library. Logs for tabular data at each iteration have been enough for everyone when using XGBoost or any other model so I don't think it worth making things more complicated.