Skip to content

Tabnet training#11

Open
hamshkhawar wants to merge 5 commits intoPolusAI:mainfrom
hamshkhawar:tabnet-training
Open

Tabnet training#11
hamshkhawar wants to merge 5 commits intoPolusAI:mainfrom
hamshkhawar:tabnet-training

Conversation

@hamshkhawar
Copy link
Member

This plugin this for training tabular data using tabNet classifer. Separate tool will be created for model prediction

torch = "2.2.2"
pytorch-tabnet = "^4.1.0"
scikit-learn = "^1.5.0"

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
numpy = "<2"

Copy link
Member

@bengtl bengtl left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some of the other dependencies require numpy < 2

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants