{
  "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"
  ]
}