Update README.md
Browse files
README.md
CHANGED
|
@@ -4,6 +4,8 @@ datasets:
|
|
| 4 |
- stanfordnlp/imdb
|
| 5 |
metrics:
|
| 6 |
- accuracy
|
|
|
|
|
|
|
| 7 |
---
|
| 8 |
|
| 9 |
# Model Card for DistilBERT Fine-Tuned on IMDb Dataset
|
|
@@ -69,4 +71,4 @@ from transformers import pipeline
|
|
| 69 |
|
| 70 |
classifier = pipeline("sentiment-analysis", model="your-model-id")
|
| 71 |
result = classifier("I loved this movie!")
|
| 72 |
-
print(result)
|
|
|
|
| 4 |
- stanfordnlp/imdb
|
| 5 |
metrics:
|
| 6 |
- accuracy
|
| 7 |
+
tags:
|
| 8 |
+
- Pytorch
|
| 9 |
---
|
| 10 |
|
| 11 |
# Model Card for DistilBERT Fine-Tuned on IMDb Dataset
|
|
|
|
| 71 |
|
| 72 |
classifier = pipeline("sentiment-analysis", model="your-model-id")
|
| 73 |
result = classifier("I loved this movie!")
|
| 74 |
+
print(result)
|