updated modeel weights

config.json

@@ -6,7 +6,6 @@
   "attention_dropout": 0.1,
   "dim": 768,
   "dropout": 0.1,
-  "dtype": "float32",
   "hidden_dim": 3072,
   "initializer_range": 0.02,
   "max_position_embeddings": 512,
@@ -18,6 +17,7 @@
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
-  "transformers_version": "4.56.2",
+  "torch_dtype": "float32",
+  "transformers_version": "4.53.2",
   "vocab_size": 30522
 }

interventions.json

@@ -1 +1,7 @@
-["Awareness/Information Provided", "Awareness/Information Provided, Counselling", "Counseling", "Counseling, Referral", "Counselling", "Counselling\nReferral", "Counselling\nReferral\nSignposting", "Counselling, Awareness/Information Provided", "Counselling, Awareness/Information Provided, Signposting", "Counselling, Referral", "Counselling, Referral, Signposting", "Counselling, Signposting", "Information Provided, Counselling", "Referral", "Referral, Counselling", "Referral, Signposting", "Signposting", "Signposting, Counselling", "Signposting, Referral"]
+[
+  "Awareness/Information Provided",
+  "Counselling",
+  "No intervention needed",
+  "Referral",
+  "Signposting"
+]

logs/run_20251001_160946/events.out.tfevents.1759324186.bitz-B760M-DS3H.2399449.0

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

main_categories.json

@@ -1 +1,9 @@
-["Advice and Counselling", "Child Maintenance & Custody", "Disability", "GBV", "Information", "Nutrition", "Unknown", "VANE"]
+[
+  "Advice and Counselling",
+  "Child Maintenance & Custody",
+  "Disability",
+  "GBV",
+  "Information",
+  "Nutrition",
+  "VANE"
+]

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8cf8c1b29afb799476d64a415f92be63ceeadaf6b9e614fa51331be557f31d9d
-size 268165304
+oid sha256:dfc065fb48eee0664be9cd861a9bf4d4f896c73e0438bc24f6fcb087f25b4656
+size 268073024

multilabel_model_metrics.json

@@ -1,32 +1,26 @@
 {
-    "version": "v-1",
-    "date_trained": "2025-10-04 11:41:45.764629",
-    "eval_avg_acc": 0.6323082763857252,
-    "last_best_model_dir": "multitask_distilbert_v-1",
-    "metrics": {
-        "eval_avg_acc": 0.6323082763857252,
-        "eval_avg_precision": 0.6088630161938874,
-        "eval_avg_recall": 0.6323082763857252,
-        "eval_avg_f1": 0.6136604194827777,
-        "eval_main_acc": 0.7205770690964313,
-        "eval_main_precision": 0.7226006095439017,
-        "eval_main_recall": 0.7205770690964313,
-        "eval_main_f1": 0.7197372354631707,
-        "eval_sub_acc": 0.5930144267274108,
-        "eval_sub_precision": 0.602177556887507,
-        "eval_sub_recall": 0.5930144267274108,
-        "eval_sub_f1": 0.5842207074361928,
-        "eval_interv_acc": 0.6324981017463933,
-        "eval_interv_precision": 0.5717722136948561,
-        "eval_interv_recall": 0.6324981017463933,
-        "eval_interv_f1": 0.5962449563397545,
-        "eval_priority_acc": 0.5831435079726651,
-        "eval_priority_precision": 0.5389016846492849,
-        "eval_priority_recall": 0.5831435079726651,
-        "eval_priority_f1": 0.5544387786919931,
-        "eval_runtime": 7.0853,
-        "eval_samples_per_second": 185.877,
-        "eval_steps_per_second": 11.714,
-        "epoch": 12.0
-    }
+    "eval_avg_acc": 0.6536885245901639,
+    "eval_avg_precision": 0.6310791053371948,
+    "eval_avg_recall": 0.6536885245901639,
+    "eval_avg_f1": 0.6310872540018378,
+    "eval_main_acc": 0.6994535519125683,
+    "eval_main_precision": 0.6950549785486344,
+    "eval_main_recall": 0.6994535519125683,
+    "eval_main_f1": 0.6942266690646444,
+    "eval_sub_acc": 0.5806010928961749,
+    "eval_sub_precision": 0.5781261111877007,
+    "eval_sub_recall": 0.5806010928961749,
+    "eval_sub_f1": 0.5734874502215787,
+    "eval_interv_acc": 0.6953551912568307,
+    "eval_interv_precision": 0.6753778417197009,
+    "eval_interv_recall": 0.6953551912568307,
+    "eval_interv_f1": 0.6720282592209204,
+    "eval_priority_acc": 0.639344262295082,
+    "eval_priority_precision": 0.5757574898927432,
+    "eval_priority_recall": 0.639344262295082,
+    "eval_priority_f1": 0.584606637500208,
+    "eval_runtime": 3.9171,
+    "eval_samples_per_second": 186.872,
+    "eval_steps_per_second": 11.743,
+    "epoch": 12.0
 }

priorities.json

@@ -1 +1,5 @@
-[1, 2, 3]
+[
+  1,
+  2,
+  3
+]

sub_categories.json

@@ -1 +1,67 @@
-["Adoption", "Albinism", "Balanced Diet", "Balanced Diet, Gender-based Violence", "Birth Registration", "Breastfeeding", "Breastfeing", "Breastfeing, Domestic Violence", "Bullying", "Child Abduction", "Child Abuse", "Child Labor", "Child Marriage", "Child Neglect", "Child Rights", "Child Trafficking", "Child in Conflict with the Law", "Custody", "Discrimination", "Drug/Alcohol Abuse", "Emotional Abuse", "Emotional/Psychological Violence", "Family Relationship", "Feeding & Food preparation", "Female Genital Mutilation", "Financial/Economic Violence", "Forced Marriage Violence", "Foster Care", "HIV/AIDS", "Harmful Practice", "Hearing impairment", "Hearing impairment, Gender based violence", "Hearing impairment, Gender-based violence", "Homelessness", "Hydrocephalus", "Info on Helpline", "Legal Issues", "Legal issues", "Maintenance", "Malnutrition", "Mental impairment", "Mental impairment, Gender based violence", "Mental impairment, Gender-based violence", "Missing Child", "Multiple disabilities", "No Care Giver", "OCSEA", "Obesity", "Other", "Outside Mandage", "Outside Mandale", "Outside Mandate", "Peer Relationships", "Physical Abuse", "Physical Health", "Physical Violence", "Physical impairment", "Psychosocial/Mental Health", "Relationships (Boy/Girl)", "Relationships (Parent/Child)", "Relationships (Student/Teacher)", "School Related Issues", "School Related Issues, Gender Based Violence", "School related issues", "School related issues, Gender based violence", "Self Esteem", "Sexual & Reproductive Health", "Sexual Abuse", "Sexual Violence", "Speech impairment", "Speech impairment, Gender based violence", "Spinal bifida", "Stagnation", "Student/ Teacher Relationship", "Teen Pregnancy", "Traditional Practice", "Traditional Practice, Gender Based Violence", "Underweight", "Unlawful Confinement", "Visual impairment"]
+[
+  "Adoption",
+  "Albinism",
+  "Balanced Diet",
+  "Birth Registration",
+  "Bullying",
+  "Child Abduction",
+  "Child Abuse",
+  "Child Labor",
+  "Child Marriage",
+  "Child Neglect",
+  "Child Rights",
+  "Child Trafficking",
+  "Child in Conflict with the Law",
+  "Custody",
+  "Discrimination",
+  "Drug/Alcohol Abuse",
+  "Emotional Abuse",
+  "Emotional/Psychological Violence",
+  "Family Relationship",
+  "Feeding & Food preparation",
+  "Female Genital Mutilation",
+  "Financial/Economic Violence",
+  "Forced Marriage Violence",
+  "Foster Care",
+  "HIV/AIDS",
+  "Harmful Practice",
+  "Hearing impairment",
+  "Homelessness",
+  "Hydrocephalus",
+  "Info on Helpline",
+  "Legal Issues",
+  "Legal issues",
+  "Maintenance",
+  "Malnutrition",
+  "Mental impairment",
+  "Missing Child",
+  "Multiple disabilities",
+  "No Care Giver",
+  "OCSEA",
+  "Obesity",
+  "Other",
+  "Peer Relationships",
+  "Physical Abuse",
+  "Physical Health",
+  "Physical Violence",
+  "Physical impairment",
+  "Psychosocial/Mental Health",
+  "Relationships (Boy/Girl)",
+  "Relationships (Parent/Child)",
+  "Relationships (Student/Teacher)",
+  "School Related Issues",
+  "School related issues",
+  "Self Esteem",
+  "Sexual & Reproductive Health",
+  "Sexual Abuse",
+  "Sexual Violence",
+  "Speech impairment",
+  "Spinal bifida",
+  "Stagnation",
+  "Student/ Teacher Relationship",
+  "Teen Pregnancy",
+  "Traditional Practice",
+  "Underweight",
+  "Unlawful Confinement",
+  "Visual impairment"
+]

tokenizer.json

@@ -2,13 +2,13 @@
   "version": "1.0",
   "truncation": {
     "direction": "Right",
-    "max_length": 256,
+    "max_length": 512,
     "strategy": "LongestFirst",
     "stride": 0
   },
   "padding": {
     "strategy": {
-      "Fixed": 256
+      "Fixed": 512
     },
     "direction": "Right",
     "pad_to_multiple_of": null,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9997d8b51d9a6e2606dafda4b46a052cb5bf8492dcca07977df4a0a4d1c80d2a
-size 5649
+oid sha256:cf30f943a10d8306a322e67590c66171ef7c118ca7717a00e977be6172ec3504
+size 5713