| { |
| "experiment_name": "DRU-RE-EntityPair-TwoHead", |
| "dataset_repo_id": "U4RASD/WojoodRelationsAnnotated", |
| "model_repo_id": "U4RASD/DRU-RE-EntityPair-TwoHead", |
| "model_name": "U4RASD/NeoAraBERT", |
| "train_file": "train.jsonl", |
| "training_eval_val_file": "val.jsonl", |
| "official_val_file": "val_official_annotated_unlabeled.jsonl", |
| "max_length": 512, |
| "context_chars": 300, |
| "num_epochs": 4.0, |
| "train_batch_size": 4, |
| "eval_batch_size": 8, |
| "grad_accum_steps": 4, |
| "effective_batch_size": 16, |
| "encoder_lr": 1e-05, |
| "head_lr": 5e-05, |
| "weight_decay": 0.01, |
| "warmup_ratio": 0.1, |
| "type_feature_mode": "one_hot_42_directional", |
| "type_feature_dim": 42, |
| "encoder_hidden_size": 768, |
| "pair_feature_dim": 3882, |
| "pair_mlp_hidden_1": 1024, |
| "pair_mlp_hidden_2": 512, |
| "pair_dropout": 0.1, |
| "existence_loss_weight": 1.0, |
| "positive_relation_loss_weight": 1.0, |
| "threshold_min": 0.01, |
| "threshold_max": 0.99, |
| "threshold_step": 0.01, |
| "tuned_existence_threshold": 0.14, |
| "official_existence_threshold_used": 0.14, |
| "official_threshold_source": "auto_tuned_on_labeled_val", |
| "best_metric": 0.5429333333333334, |
| "best_checkpoint": "outputs/DRU-RE-EntityPair-TwoHead/checkpoints/checkpoint-1500", |
| "labels": [ |
| "Administration.leader_of", |
| "Administration.manager_of", |
| "Administration.president_of", |
| "Affiliation.employee_of", |
| "Affiliation.member_of", |
| "Affiliation.owner_of", |
| "Affiliation.student_at", |
| "Business.has_competitor", |
| "Business.has_conflict_with", |
| "Business.has_partner_with", |
| "Family.has_parent", |
| "Family.has_relative", |
| "Family.has_sibling", |
| "Family.has_spouse", |
| "GPE.capital_of", |
| "GPE.has_area", |
| "GPE.has_currency", |
| "GPE.has_population", |
| "GPE.official_language", |
| "Location.has_border_with", |
| "Location.headquartered_in", |
| "Location.lives_in", |
| "Location.located_in", |
| "Location.nearby", |
| "Organization.branch_count", |
| "Organization.employs", |
| "Organization.found_on", |
| "Organization.has_alternate_name", |
| "Organization.has_propoerty", |
| "Organization.has_revenue", |
| "PartOf.geopolitical_division", |
| "PartOf.subsidiary", |
| "Personal.birth_date", |
| "Personal.birth_place", |
| "Personal.death_date", |
| "Personal.has_occupation", |
| "Productivity.builder_of", |
| "Productivity.founder_of", |
| "Productivity.inventor_of", |
| "Productivity.manufacturer_of", |
| "no_relation" |
| ], |
| "positive_labels": [ |
| "Administration.leader_of", |
| "Administration.manager_of", |
| "Administration.president_of", |
| "Affiliation.employee_of", |
| "Affiliation.member_of", |
| "Affiliation.owner_of", |
| "Affiliation.student_at", |
| "Business.has_competitor", |
| "Business.has_conflict_with", |
| "Business.has_partner_with", |
| "Family.has_parent", |
| "Family.has_relative", |
| "Family.has_sibling", |
| "Family.has_spouse", |
| "GPE.capital_of", |
| "GPE.has_area", |
| "GPE.has_currency", |
| "GPE.has_population", |
| "GPE.official_language", |
| "Location.has_border_with", |
| "Location.headquartered_in", |
| "Location.lives_in", |
| "Location.located_in", |
| "Location.nearby", |
| "Organization.branch_count", |
| "Organization.employs", |
| "Organization.found_on", |
| "Organization.has_alternate_name", |
| "Organization.has_propoerty", |
| "Organization.has_revenue", |
| "PartOf.geopolitical_division", |
| "PartOf.subsidiary", |
| "Personal.birth_date", |
| "Personal.birth_place", |
| "Personal.death_date", |
| "Personal.has_occupation", |
| "Productivity.builder_of", |
| "Productivity.founder_of", |
| "Productivity.inventor_of", |
| "Productivity.manufacturer_of" |
| ], |
| "entity_types": [ |
| "CARDINAL", |
| "CURR", |
| "DATE", |
| "EVENT", |
| "FAC", |
| "GPE", |
| "LANGUAGE", |
| "LAW", |
| "LOC", |
| "MONEY", |
| "NORP", |
| "OCC", |
| "ORDINAL", |
| "ORG", |
| "PERCENT", |
| "PERS", |
| "PRODUCT", |
| "QUANTITY", |
| "TIME", |
| "UNIT", |
| "WEBSITE" |
| ] |
| } |
