thoriqsaputra commited on
Commit
00491de
·
verified ·
1 Parent(s): 9df7e8e

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +25 -0
README.md ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - text-classification
4
+ - sentiment-analysis
5
+ - distilbert
6
+ license: apache-2.0
7
+ ---
8
+
9
+ # My Awesome Model
10
+
11
+ This is a fine-tuned DistilBERT model for sentiment analysis on the IMDb dataset. It classifies movie reviews as positive or negative.
12
+
13
+ ## Usage
14
+
15
+ ```python
16
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
17
+ model = AutoModelForSequenceClassification.from_pretrained("your_username/testing_the_water")
18
+ tokenizer = AutoTokenizer.from_pretrained("your_username/testing_the_water")
19
+ ```
20
+
21
+ ## Training
22
+
23
+ - Dataset: IMDb (subset of 1,000 samples)
24
+ - Epochs: 1
25
+ - Batch Size: 8