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:
|