Hugging Face's logo

oussama120
/
Resume_Sentence_Classification

Text Classification
Transformers
PyTorch
distilbert
fine-tuning
resume classification
text-embeddings-inference
Model card Files
xet
 Community
1
oussama120 commited on
Commit
333c786
·
verified ·
1 Parent(s): 4690d80

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -3
README.md CHANGED
@@ -1,3 +1,10 @@
 
 
 
 
 
 
 
1
  # DistilBERT Resume Classification Model
2
 
3
  This repository contains a fine-tuned DistilBERT model for classifying resume sentences into predefined categories. The model is trained on a dataset of resumes and can classify sentences into categories such as Personal Information, Experience, Summary, Education, Qualifications & Certificates, Skills, and Objectives.
@@ -31,6 +38,4 @@ from transformers import DistilBertTokenizerFast, DistilBertForSequenceClassific
31
  # Load the model and tokenizer
32
  model_name = "oussama120/Resume_Sentence_Classification"
33
  tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
34
- model = DistilBertForSequenceClassification.from_pretrained(model_name)
35
-
36
-
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - text-classification
5
+ - fine-tuning
6
+ - resume classification
7
+ ---
8
  # DistilBERT Resume Classification Model
9
 
10
  This repository contains a fine-tuned DistilBERT model for classifying resume sentences into predefined categories. The model is trained on a dataset of resumes and can classify sentences into categories such as Personal Information, Experience, Summary, Education, Qualifications & Certificates, Skills, and Objectives.
 
38
  # Load the model and tokenizer
39
  model_name = "oussama120/Resume_Sentence_Classification"
40
  tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
41
+ model = DistilBertForSequenceClassification.from_pretrained(model_name)