Update README.md
Browse files
README.md
CHANGED
|
@@ -54,6 +54,7 @@ print(probs.argmax(dim=-1))
|
|
| 54 |
```
|
| 55 |
|
| 56 |
You can convert the `logits` of the model with a softmax to obtain a probability distribution over the following 24 categories (in order of labels, also see `id2label` and `label2id` in the model config):
|
|
|
|
| 57 |
0. Academic Writing
|
| 58 |
1. Content Listing
|
| 59 |
2. Creative Writing
|
|
|
|
| 54 |
```
|
| 55 |
|
| 56 |
You can convert the `logits` of the model with a softmax to obtain a probability distribution over the following 24 categories (in order of labels, also see `id2label` and `label2id` in the model config):
|
| 57 |
+
|
| 58 |
0. Academic Writing
|
| 59 |
1. Content Listing
|
| 60 |
2. Creative Writing
|