FDSRashid commited on
Commit
1b12e93
ยท
verified ยท
1 Parent(s): a96cd50

added narrator detection model to ukhbert

Browse files
config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertForTokenClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "directionality": "bidi",
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3",
16
+ "4": "LABEL_4"
17
+ },
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "label2id": {
21
+ "LABEL_0": 0,
22
+ "LABEL_1": 1,
23
+ "LABEL_2": 2,
24
+ "LABEL_3": 3,
25
+ "LABEL_4": 4
26
+ },
27
+ "layer_norm_eps": 1e-12,
28
+ "max_position_embeddings": 512,
29
+ "model_type": "bert",
30
+ "num_attention_heads": 12,
31
+ "num_hidden_layers": 12,
32
+ "pad_token_id": 0,
33
+ "pooler_fc_size": 768,
34
+ "pooler_num_attention_heads": 12,
35
+ "pooler_num_fc_layers": 3,
36
+ "pooler_size_per_head": 128,
37
+ "pooler_type": "first_token_transform",
38
+ "type_vocab_size": 2,
39
+ "vocab_size": 119547
40
+ }
eval_results.txt ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ acc = 0.9886209625807463
2
+ correct_hadith = 0.7687676387582314
3
+ eval_loss = 0.05440825115018377
4
+ f1_score = 0.9607255174431361
5
+ precision = 0.9398847389714969
6
+ recall = 0.9825114912829439
model_args.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"adam_epsilon": 1e-08, "best_model_dir": "outputs/best_model", "cache_dir": "cache_dir/", "config": {}, "custom_layer_parameters": [], "custom_parameter_groups": [], "dataloader_num_workers": 30, "do_lower_case": false, "dynamic_quantize": false, "early_stopping_consider_epochs": false, "early_stopping_delta": 0, "early_stopping_metric": "eval_loss", "early_stopping_metric_minimize": true, "early_stopping_patience": 3, "encoding": null, "eval_batch_size": 8, "evaluate_during_training": true, "evaluate_during_training_silent": true, "evaluate_during_training_steps": 0, "evaluate_during_training_verbose": true, "fp16": false, "gradient_accumulation_steps": 1, "learning_rate": 4e-05, "local_rank": -1, "logging_steps": 50, "manual_seed": null, "max_grad_norm": 1.0, "max_seq_length": 512, "model_name": "updated_RD2/checkpoint-152043-epoch-2", "model_type": "bert", "multiprocessing_chunksize": 500, "n_gpu": 1, "no_cache": false, "no_save": false, "num_train_epochs": 5, "output_dir": "outputs/", "overwrite_output_dir": false, "process_count": 30, "reprocess_input_data": true, "save_best_model": true, "save_eval_checkpoints": false, "save_model_every_epoch": true, "save_optimizer_and_scheduler": true, "save_steps": 0, "silent": false, "tensorboard_dir": null, "thread_count": null, "train_batch_size": 8, "train_custom_parameters_only": false, "use_cached_eval_features": false, "use_early_stopping": false, "use_multiprocessing": true, "wandb_kwargs": {}, "wandb_project": null, "warmup_ratio": 0.06, "warmup_steps": 15205, "weight_decay": 0, "model_class": "NERModel", "classification_report": false, "labels_list": ["[CLS]", "[SEP]", "O", "B-PER", "I-PER"], "lazy_loading": false, "lazy_loading_start_line": 0}
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:265d0db80920fdca4e48d87dd0a20534e00b7c2fe36e328a75f6c1c241d45cf8
3
+ size 1418189403
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a78ded39030ddde693113c9cc28b5552d13ccc7abf6a2c00ef0d085236e16438
3
+ size 709154122
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee6038ace1bfe595ad7d5e3fbd8cd1ae7f76c4528ae60f00f5a0a2b7445b36f1
3
+ size 330
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"do_lower_case": false, "model_max_length": 512, "special_tokens_map_file": "updated_RD/checkpoint-50681-epoch-1/special_tokens_map.json", "full_tokenizer_file": null}
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6abda986c66dd2dd56420aa4edd75177b885830bebd40152fb2951f87d663ed
3
+ size 1984
vocab.txt ADDED
The diff for this file is too large to render. See raw diff