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,
"hidden_dim": 3072,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1",
"2": "LABEL_2"
"0": "B-KEY",
"1": "I-KEY",
"2": "O"
},
"initializer_range": 0.02,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_2": 2
"B-KEY": 0,
"I-KEY": 1,
"O": 2
},
"max_position_embeddings": 512,
"model_type": "distilbert",