vitiligo-relation-extractor / relations.json
re-cep's picture
feat(relations): upgrade to ATLOP context pooling, between-span features, and domain-prior masking
c0fb351 verified
Raw History Blame Contribute Delete
2.87 kB
{
"base_model": "microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext",
"bias": 0.4,
"labels": [
"None",
"Association",
"Positive_Correlation",
"Negative_Correlation",
"Bind",
"Cotreatment",
"Comparison",
"Drug_Interaction",
"Conversion"
],
"special": [
2,
3,
0
],
"history": [
{
"epoch": 1,
"seconds": 86,
"bias": -1.0,
"precision": 0.5272255834053586,
"recall": 0.5249569707401033,
"f1": 0.5260888313928417,
"correct": 610,
"predicted": 1157,
"gold": 1162
},
{
"epoch": 2,
"seconds": 82,
"bias": 0.0,
"precision": 0.5887850467289719,
"recall": 0.5421686746987951,
"f1": 0.564516129032258,
"correct": 630,
"predicted": 1070,
"gold": 1162
},
{
"epoch": 3,
"seconds": 79,
"bias": -0.1,
"precision": 0.5911151405258386,
"recall": 0.5611015490533563,
"f1": 0.5757174392935983,
"correct": 652,
"predicted": 1103,
"gold": 1162
},
{
"epoch": 4,
"seconds": 79,
"bias": -0.9,
"precision": 0.5614617940199336,
"recall": 0.5817555938037866,
"f1": 0.5714285714285714,
"correct": 676,
"predicted": 1204,
"gold": 1162
},
{
"epoch": 5,
"seconds": 79,
"bias": 0.4,
"precision": 0.633175355450237,
"recall": 0.5748709122203098,
"f1": 0.602616147947677,
"correct": 668,
"predicted": 1055,
"gold": 1162
},
{
"epoch": 6,
"seconds": 80,
"bias": -0.2,
"precision": 0.6325358851674641,
"recall": 0.5688468158347676,
"f1": 0.5990031717263253,
"correct": 661,
"predicted": 1045,
"gold": 1162
},
{
"epoch": 7,
"seconds": 78,
"bias": -0.9,
"precision": 0.5961214165261383,
"recall": 0.608433734939759,
"f1": 0.60221465076661,
"correct": 707,
"predicted": 1186,
"gold": 1162
},
{
"epoch": 8,
"seconds": 77,
"bias": -1.2,
"precision": 0.6260704091341579,
"recall": 0.5662650602409639,
"f1": 0.5946678716674199,
"correct": 658,
"predicted": 1051,
"gold": 1162
},
{
"epoch": 9,
"seconds": 77,
"bias": -0.7,
"precision": 0.6156615661566157,
"recall": 0.5886402753872634,
"f1": 0.6018477782666081,
"correct": 684,
"predicted": 1111,
"gold": 1162
},
{
"epoch": 10,
"seconds": 77,
"bias": 0.1,
"precision": 0.6510204081632653,
"recall": 0.5490533562822719,
"f1": 0.5957049486461251,
"correct": 638,
"predicted": 980,
"gold": 1162
}
],
"silver": false,
"init": "data/models/relations/biomedbert-silver.pt"
}