osliusarenko
commited on
Commit
•
0e7a564
1
Parent(s):
c6ec0d3
Update README.md
Browse files
README.md
CHANGED
@@ -12,6 +12,14 @@ This is a baseline RoBERTa-base model for the delicate text detection task.
|
|
12 |
* Paper: [DeTexD: A Benchmark Dataset for Delicate Text Detection](TODO)
|
13 |
* [GitHub repository](https://github.com/grammarly/detexd)
|
14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
## Classification example code
|
16 |
|
17 |
Here's a short usage example with the torch library in a binary classification task:
|
|
|
12 |
* Paper: [DeTexD: A Benchmark Dataset for Delicate Text Detection](TODO)
|
13 |
* [GitHub repository](https://github.com/grammarly/detexd)
|
14 |
|
15 |
+
The labels meaning according to the paper:
|
16 |
+
- LABEL_0 -> non-delicate (0)
|
17 |
+
- LABEL_1 -> very low risk (1)
|
18 |
+
- LABEL_2 -> low risk (2)
|
19 |
+
- LABEL_3 -> medium risk (3)
|
20 |
+
- LABEL_4 -> high risk (4)
|
21 |
+
- LABEL_5 -> very high risk (5)
|
22 |
+
|
23 |
## Classification example code
|
24 |
|
25 |
Here's a short usage example with the torch library in a binary classification task:
|