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

GH-748: add more classification datasets #749

Merged
merged 1 commit into from
May 24, 2019

Conversation

alanakbik
Copy link
Collaborator

This PR adds more datasets to the new flair.datasets way of loading data. It adds:

  • IMDB dataset
  • TREC_50 dataset

It also adds the popular 20newsgroups classification dataset to Flair.

Load with:

import flair.datasets

# load IMDB dataset
imdb_corpus = flair.datasets.IMDB()
print(imdb_corpus)

# load 20 newsgroups dataset
newsgroup_corpus = flair.datasets.NEWSGROUPS()
print(newsgroup_corpus) 

@alanakbik
Copy link
Collaborator Author

👍

1 similar comment
@kashif
Copy link
Contributor

kashif commented May 24, 2019

👍

@kashif kashif merged commit 7cd9b89 into master May 24, 2019
@alanakbik alanakbik deleted the GH-748-classification-datasets branch May 28, 2019 12:09
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