osliusarenko commited on
Commit
0e7a564
1 Parent(s): c6ec0d3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -0
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: