# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("vedalken/ML6-interview")
model = AutoModelForSequenceClassification.from_pretrained("vedalken/ML6-interview")Quick Links
Roberta-base trained on the dataset for job description classification.
Repo of the interview: https://bitbucket.org/ml6team/challenge-classify-job-descriptions.git/src
For the training code just send me a message or comment in this repo.
- Downloads last month
- 2
Model tree for vedalken/ML6-interview
Base model
FacebookAI/roberta-base
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="vedalken/ML6-interview")