Update README.md
Browse files
README.md
CHANGED
|
@@ -19,7 +19,7 @@ model-index:
|
|
| 19 |
metrics:
|
| 20 |
- name: Accuracy
|
| 21 |
type: accuracy
|
| 22 |
-
value: 0.
|
| 23 |
---
|
| 24 |
|
| 25 |
# Model Card for DistilBERT Fine-Tuned on IMDb Dataset
|
|
@@ -83,6 +83,6 @@ Use the code below to get started with the model:
|
|
| 83 |
```python
|
| 84 |
from transformers import pipeline
|
| 85 |
|
| 86 |
-
classifier = pipeline("sentiment-analysis", model="
|
| 87 |
result = classifier("I loved this movie!")
|
| 88 |
print(result)
|
|
|
|
| 19 |
metrics:
|
| 20 |
- name: Accuracy
|
| 21 |
type: accuracy
|
| 22 |
+
value: 0.9316
|
| 23 |
---
|
| 24 |
|
| 25 |
# Model Card for DistilBERT Fine-Tuned on IMDb Dataset
|
|
|
|
| 83 |
```python
|
| 84 |
from transformers import pipeline
|
| 85 |
|
| 86 |
+
classifier = pipeline("sentiment-analysis", model="3oclock/distilbert-imdb")
|
| 87 |
result = classifier("I loved this movie!")
|
| 88 |
print(result)
|