ml6team
/
keyphrase-extraction-distilbert-inspec
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update labels

This commit is contained in:
Thomas De Decker 2022-05-04 15:38:37 +02:00
parent 7d176e7f1f
commit 514f1dc492
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
config.json
View File

@ -9,15 +9,15 @@
"dropout": 0.1, "dropout": 0.1,
"hidden_dim": 3072, "hidden_dim": 3072,
"id2label": { "id2label": {
"0": "LABEL_0", "0": "B-KEY",
"1": "LABEL_1", "1": "I-KEY",
"2": "LABEL_2" "2": "O"
}, },
"initializer_range": 0.02, "initializer_range": 0.02,
"label2id": { "label2id": {
"LABEL_0": 0, "B-KEY": 0,
"LABEL_1": 1, "I-KEY": 1,
"LABEL_2": 2 "O": 2
}, },
"max_position_embeddings": 512, "max_position_embeddings": 512,
"model_type": "distilbert", "model_type": "distilbert",