diff --git a/config.json b/config.json index 0d20405..6ebbbad 100644 --- a/config.json +++ b/config.json @@ -12,19 +12,19 @@ "hidden_size": 1024, "id2label": { "0": "O", - "1": "LOC", - "2": "PER", - "3": "MISC", - "4": "ORG" + "1": "PER", + "2": "ORG", + "3": "LOC", + "4": "MISC" }, "initializer_range": 0.02, "intermediate_size": 4096, "label2id": { - "LOC": 1, - "MISC": 3, + "LOC": 3, + "MISC": 4, "O": 0, - "ORG": 4, - "PER": 2 + "ORG": 2, + "PER": 1 }, "layer_norm_eps": 1e-05, "max_position_embeddings": 514, diff --git a/pytorch_model.bin b/pytorch_model.bin index def7382..e9861cc 100644 --- a/pytorch_model.bin +++ b/pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:0e77a9cef4873df5643217b672929b3f8d3113b4a177bf593096d7b9db7e03f4 +oid sha256:689bbe69ce7fb4aedd7dbe43be806cb67e4e2d512ffa5de81d1ab0c89e87d654 size 1417433007 diff --git a/results.csv b/results.csv index 3e3a0ba..7cdf4dd 100644 --- a/results.csv +++ b/results.csv @@ -1,6 +1,6 @@ ,precision,recall,f1,entity -0,0.9795249795249795,0.9862561847168774,0.9828790576633339,LOC -1,0.9914318668643928,0.9927404718693285,0.9920857378400659,PER -2,0.9292274446245273,0.9262250942380184,0.9277238403451995,MISC -3,0.9627007895453308,0.966120218579235,0.9644074730669576,ORG -4,0.9740825890497252,0.9766692954784437,0.9753719894698967,Overall +0,0.9904511030622325,0.9925754825936314,0.9915121549237741,PER +1,0.9628323385784048,0.969672131147541,0.966240130683365,ORG +2,0.974924221548636,0.9725123694337549,0.9737168019815605,LOC +3,0.9308278867102396,0.9203015616585891,0.925534795559166,MISC +4,0.9728188879121981,0.9734490010515248,0.9731265700746845,Overall