Update README.md
Browse files
README.md
CHANGED
|
@@ -54,30 +54,30 @@ 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 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
|
| 81 |
|
| 82 |
The full definitions of the categories can be found in the [taxonomy config](https://github.com/CodeCreator/WebOrganizer/blob/main/define_domains/taxonomies/topics.yaml).
|
| 83 |
|
|
|
|
| 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
|
| 60 |
+
3. Customer Support
|
| 61 |
+
4. Comment Section
|
| 62 |
+
5. FAQ
|
| 63 |
+
6. Truncated
|
| 64 |
+
7. Knowledge Article
|
| 65 |
+
8. Legal Notices
|
| 66 |
+
9. Listicle
|
| 67 |
+
10. News Article
|
| 68 |
+
11. Nonfiction Writing
|
| 69 |
+
12. About (Org
|
| 70 |
+
13. News (Org
|
| 71 |
+
14. About (Pers
|
| 72 |
+
15. Personal Blog
|
| 73 |
+
16. Product Page
|
| 74 |
+
17. Q&A Forum
|
| 75 |
+
18. Spam / Ads
|
| 76 |
+
19. Structured Data
|
| 77 |
+
20. Documentation
|
| 78 |
+
21. Audio Transcript
|
| 79 |
+
22. Tutorial
|
| 80 |
+
23. User Review
|
| 81 |
|
| 82 |
The full definitions of the categories can be found in the [taxonomy config](https://github.com/CodeCreator/WebOrganizer/blob/main/define_domains/taxonomies/topics.yaml).
|
| 83 |
|