# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("daspartho/subreddit-predictor")
model = AutoModelForSequenceClassification.from_pretrained("daspartho/subreddit-predictor")Quick Links
An NLP model that predicts subreddit based on the title of a post.
Training
DistilBERT is fine-tuned on subreddit-posts, a dataset of titles of the top 1000 posts from the top 250 subreddits.
For steps to make the model check out the model notebook in the github repo or open in Colab.
Limitations and bias
- Since the model is trained on top 250 subreddits (for reference) therefore it can only categorise within those subreddits.
- Some subreddits have a specific format for their post title, like r/todayilearned where post title starts with "TIL" so the model becomes biased towards "TIL" --> r/todayilearned. This can be removed by cleaning the dataset of these specific terms.
- In some subreddit like r/gifs, the title of the post doesn't matter much, so the model performs poorly on them.
- Downloads last month
- 8
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="daspartho/subreddit-predictor")