Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add TabNet Datasets #1153

Merged
merged 9 commits into from
Apr 23, 2021
Merged

Add TabNet Datasets #1153

merged 9 commits into from
Apr 23, 2021

Conversation

kanishk16
Copy link
Contributor

@kanishk16 kanishk16 commented Apr 14, 2021

This PR updates the KaggleDownloadMixin and adds the following datasets from the TabNet paper:

  • Poker Hand
  • Rossmann Store Sales
  • Sarcos
  • Adult Census Income
  • Mushroom Edibility
  • Forest cover
  • KDD 2009 Churn
  • KDD 2009 Upsell
  • KDD 2009 Appetency

ludwig/datasets/mixins/process.py Outdated Show resolved Hide resolved
ludwig/datasets/sarcos/config.yaml Outdated Show resolved Hide resolved
ludwig/datasets/mixins/load.py Outdated Show resolved Hide resolved
ludwig/datasets/mixins/kaggle.py Show resolved Hide resolved
ludwig/datasets/goemotions/__init__.py Show resolved Hide resolved
@w4nderlust w4nderlust merged commit 6e14a6e into ludwig-ai:master Apr 23, 2021
@kanishk16 kanishk16 deleted the add/datasets branch April 26, 2021 15:51
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.

None yet

2 participants