init

config.json

@@ -0,0 +1,42 @@
+{
+  "_name_or_path": "xlm-roberta-base",
+  "architectures": [
+    "XLMRobertaForTokenClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "bos_token_id": 0,
+  "eos_token_id": 2,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "O",
+    "1": "B-location",
+    "2": "I-location",
+    "3": "B-organization",
+    "4": "I-organization",
+    "5": "B-person",
+    "6": "I-person"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "B-location": 1,
+    "B-organization": 3,
+    "B-person": 5,
+    "I-location": 2,
+    "I-organization": 4,
+    "I-person": 6,
+    "O": 0
+  },
+  "layer_norm_eps": 1e-05,
+  "max_position_embeddings": 514,
+  "model_type": "xlm-roberta",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "output_past": true,
+  "pad_token_id": 1,
+  "type_vocab_size": 1,
+  "vocab_size": 250002
+}

parameter.json

@@ -0,0 +1 @@
+{"dataset": ["panx_dataset/ar"], "transformers_model": "xlm-roberta-base", "random_seed": 1234, "lr": 1e-05, "total_step": 13000, "warmup_step": 700, "weight_decay": 1e-07, "batch_size": 16, "max_seq_length": 128, "fp16": false, "max_grad_norm": 1.0, "lower_case": false}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:034eaa0c81a6aae292e5cdece632edf1f30f5674829eb3f44622dc3193ab9fd5
+size 1109918343

sentencepiece.bpe.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cfc8146abe2a0488e9e2a0c56de7952f7c11ab059eca145a0a727afce0db2865
+size 5069051

special_tokens_map.json

@@ -0,0 +1 @@
+{"bos_token": "<s>", "eos_token": "</s>", "unk_token": "<unk>", "sep_token": "</s>", "pad_token": "<pad>", "cls_token": "<s>", "mask_token": "<mask>"}

test_panx_dataset-ar.json

@@ -0,0 +1 @@
+{"valid": {"f1": 89.79681978798585, "recall": 90.21922428330524, "precision": 89.37835223775609, "summary": "              precision    recall  f1-score   support\n\n    location       0.90      0.92      0.91      3856\norganization       0.86      0.87      0.86      3596\n      person       0.91      0.92      0.92      3815\n\n   micro avg       0.89      0.90      0.90     11267\n   macro avg       0.89      0.90      0.90     11267\nweighted avg       0.89      0.90      0.90     11267\n"}, "test": {"f1": 88.91536492807343, "recall": 89.48396838085088, "precision": 88.35394194510216, "summary": "              precision    recall  f1-score   support\n\n    location       0.90      0.91      0.90      3780\norganization       0.86      0.86      0.86      3629\n      person       0.90      0.92      0.91      3850\n\n   micro avg       0.88      0.89      0.89     11259\n   macro avg       0.88      0.89      0.89     11259\nweighted avg       0.88      0.89      0.89     11259\n"}}

test_panx_dataset-en.json

@@ -0,0 +1 @@
+{"valid": {"f1": 54.08364039637963, "recall": 53.55295784626284, "precision": 54.62494580141639, "summary": "              precision    recall  f1-score   support\n\n    location       0.47      0.49      0.48      4803\norganization       0.43      0.38      0.40      4677\n      person       0.74      0.74      0.74      4635\n\n   micro avg       0.55      0.54      0.54     14115\n   macro avg       0.54      0.54      0.54     14115\nweighted avg       0.54      0.54      0.54     14115\n"}, "test": {"f1": 53.107013234593246, "recall": 52.13041600690945, "precision": 54.120899648808184, "summary": "              precision    recall  f1-score   support\n\n    location       0.43      0.44      0.44      4633\norganization       0.45      0.39      0.42      4744\n      person       0.74      0.74      0.74      4517\n\n   micro avg       0.54      0.52      0.53     13894\n   macro avg       0.54      0.52      0.53     13894\nweighted avg       0.54      0.52      0.53     13894\n"}}

test_panx_dataset-es.json

@@ -0,0 +1 @@
+{"valid": {"f1": 72.27661927977384, "recall": 71.89079054604727, "precision": 72.66661174726089, "summary": "              precision    recall  f1-score   support\n\n    location       0.76      0.71      0.74      4567\norganization       0.59      0.63      0.61      3737\n      person       0.82      0.81      0.81      3966\n\n   micro avg       0.73      0.72      0.72     12270\n   macro avg       0.73      0.72      0.72     12270\nweighted avg       0.73      0.72      0.72     12270\n"}, "test": {"f1": 73.61082632766046, "recall": 73.20554649265905, "precision": 74.02061855670104, "summary": "              precision    recall  f1-score   support\n\n    location       0.79      0.72      0.76      4725\norganization       0.59      0.64      0.62      3576\n      person       0.83      0.82      0.83      3959\n\n   micro avg       0.74      0.73      0.74     12260\n   macro avg       0.74      0.73      0.73     12260\nweighted avg       0.75      0.73      0.74     12260\n"}}

test_panx_dataset-ja.json

@@ -0,0 +1 @@
+{"valid": {"f1": 49.21174761752932, "recall": 53.96554068491288, "precision": 45.227668614778516, "summary": "              precision    recall  f1-score   support\n\n    location       0.47      0.56      0.51     19595\norganization       0.34      0.44      0.38     16328\n      person       0.57      0.62      0.60     15558\n\n   micro avg       0.45      0.54      0.49     51481\n   macro avg       0.46      0.54      0.50     51481\nweighted avg       0.46      0.54      0.50     51481\n"}, "test": {"f1": 49.21174761752932, "recall": 53.96554068491288, "precision": 45.227668614778516, "summary": "              precision    recall  f1-score   support\n\n    location       0.47      0.56      0.51     19595\norganization       0.34      0.44      0.38     16328\n      person       0.57      0.62      0.60     15558\n\n   micro avg       0.45      0.54      0.49     51481\n   macro avg       0.46      0.54      0.50     51481\nweighted avg       0.46      0.54      0.50     51481\n"}}

test_panx_dataset-ko.json

@@ -0,0 +1 @@
+{"valid": {"f1": 59.8928668294006, "recall": 60.67871127292711, "precision": 59.12711694223174, "summary": "              precision    recall  f1-score   support\n\n    location       0.65      0.64      0.64      5997\norganization       0.49      0.49      0.49      4459\n      person       0.62      0.68      0.65      4101\n\n   micro avg       0.59      0.61      0.60     14557\n   macro avg       0.58      0.60      0.59     14557\nweighted avg       0.59      0.61      0.60     14557\n"}, "test": {"f1": 59.694485081481986, "recall": 60.82645774110795, "precision": 58.60387441549766, "summary": "              precision    recall  f1-score   support\n\n    location       0.64      0.65      0.65      5855\norganization       0.48      0.48      0.48      4319\n      person       0.61      0.67      0.64      4249\n\n   micro avg       0.59      0.61      0.60     14423\n   macro avg       0.58      0.60      0.59     14423\nweighted avg       0.58      0.61      0.60     14423\n"}}

test_panx_dataset-ru.json

@@ -0,0 +1 @@
+{"valid": {"f1": 49.94147768836869, "recall": 55.351062104751094, "precision": 45.49513527922164, "summary": "              precision    recall  f1-score   support\n\n    location       0.51      0.77      0.62      4852\norganization       0.60      0.56      0.58      3892\n      person       0.22      0.25      0.23      3590\n\n   micro avg       0.45      0.55      0.50     12334\n   macro avg       0.44      0.53      0.48     12334\nweighted avg       0.46      0.55      0.49     12334\n"}, "test": {"f1": 49.34978572484114, "recall": 54.84930606881826, "precision": 44.85259552749984, "summary": "              precision    recall  f1-score   support\n\n    location       0.50      0.77      0.61      4560\norganization       0.60      0.58      0.59      4074\n      person       0.21      0.24      0.22      3543\n\n   micro avg       0.45      0.55      0.49     12177\n   macro avg       0.44      0.53      0.47     12177\nweighted avg       0.45      0.55      0.49     12177\n"}}

tokenizer_config.json

@@ -0,0 +1 @@
+{"bos_token": "<s>", "eos_token": "</s>", "sep_token": "</s>", "cls_token": "<s>", "unk_token": "<unk>", "pad_token": "<pad>", "mask_token": "<mask>", "model_max_length": 512, "name_or_path": "xlm-roberta-base"}