insikt commited on
Commit
9c6cdc4
·
1 Parent(s): e97e7b6

First version of the ES-threat model

Browse files

The first version of the threat model was trained in the Spanish dataset on top of the xml-Roberta model.

Files changed (1) hide show
  1. README.md +22 -0
README.md CHANGED
@@ -1,3 +1,25 @@
1
  ---
2
  license: openrail
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: openrail
3
+ language:
4
+ - es
5
+ metrics:
6
+ - accuracy
7
+ pipeline_tag: text-classification
8
  ---
9
+
10
+ Text classification model for threat detection in Spanish.
11
+
12
+ We define threat messages as messages containing threats, support of violence, or harm towards a person or group of people.
13
+
14
+ It is trained on the XML-RobertA (xlm-roberta-base) model with a dataset of 364.793 messages from online sources.
15
+
16
+ Parameters
17
+ "learning_rate": 1e-5,
18
+ "weight_decay": 0.01,
19
+ "opimizer": "Adam",
20
+ "eps": 1e-08,
21
+ "num_labels": 2,
22
+ "output_attentions": "True",
23
+ "output_hidden_states": "False",
24
+ "number_epoch": 3,
25
+ "batch-size": 16