Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
1 |
# bert-sentiment-imdb
|
2 |
|
3 |
This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset.
|
@@ -8,4 +13,4 @@ This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset.
|
|
8 |
- **Usage**:
|
9 |
```python
|
10 |
from transformers import pipeline
|
11 |
-
classifier = pipeline("sentiment-analysis", model="zavora/bert-sentiment-imdb")
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
base_model:
|
4 |
+
- google-bert/bert-base-uncased
|
5 |
+
---
|
6 |
# bert-sentiment-imdb
|
7 |
|
8 |
This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset.
|
|
|
13 |
- **Usage**:
|
14 |
```python
|
15 |
from transformers import pipeline
|
16 |
+
classifier = pipeline("sentiment-analysis", model="zavora/bert-sentiment-imdb")
|