Minghui131 commited on
Commit
3b439cb
1 Parent(s): 541c66c

Create README.md

Browse files

## IMDB Sentiment Classifier

This model is a sentiment classifier trained on the IMDB dataset. It uses a transformer-based architecture and was fine-tuned on a pre-trained BERT model.

### Model Details
- **Model Architecture:** BERT-base
- **Training Data:** IMDB movie reviews dataset
- **Training Procedure:** Fine-tuning on BERT base model with a learning rate of 2e-5 for 3 epochs.
- **Evaluation Metrics:** Accuracy, F1 Score

Files changed (1) hide show
  1. README.md +12 -0
README.md ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets:
4
+ - mteb/imdb
5
+ language:
6
+ - en
7
+ metrics:
8
+ - accuracy
9
+ library_name: adapter-transformers
10
+ tags:
11
+ - art
12
+ ---