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.55k stars 470 forks source link

saving/loading tabnet classifier model #452

Closed maymag closed 1 year ago

maymag commented 1 year ago

Describe the bug

What is the current behavior? I'm unable to load a trained TabNetClassifier() saved model. I used a custom evaluation metric during training. The model was "Successfully saved". There is no error when I try to load. It just loads nothing. I saved and loaded according to documentation. I

If the current behavior is a bug, please provide the steps to reproduce.

Expected behavior It should load a model but it's not.

Screenshots

Other relevant information: poetry version:
python version: Python 3.7.12 Operating System: "Ubuntu 20.04.4 LTS" Additional tools:

Additional context

Optimox commented 1 year ago

Hello, I will definitely need more information in order to help you.

Can you share a minimal reproducible code and the error message you encounter ?