Upload trained model (Safetensors) Acc: 73.58%

README.md

@@ -19,6 +19,6 @@ Biased Activation Detection (BAD) classifier for TinyLlama-1.1B.
 - **Config**: `config.json`
 
 ## Stats
-- **Balanced Accuracy**: 74.51%
+- **Balanced Accuracy**: 73.58%
 - **Best Layer**: 17
 - **Training Date**: 2025-12-12

config.json

@@ -2,10 +2,10 @@
   "input_dim": 2048,
   "layer_idx": 17,
   "base_model": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
-  "best_val_bal_acc": 0.7450792126740279,
+  "best_val_bal_acc": 0.7358173076923077,
   "training_method": "GroupSplit + Standardized + Balanced",
-  "training_date": "2025-12-12T09:10:24.028473",
-  "dropout_rate": 0.3,
-  "hidden_dim": null,
-  "architecture": "linear"
+  "training_date": "2025-12-12T16:56:48.629064",
+  "dropout_rate": 0.1,
+  "hidden_dim": 256,
+  "architecture": "hidden_layer"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:51e1f78f2c7a243092aec3b75c32f3df29e69522f25f62b3ecc26cd92b4324ab
-size 10523572
+oid sha256:fb4e060ac314e8ba93295ecbfc80b987eda558280fbb9553400f28a234d0f951
+size 2099508

scaler.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ea72ab8211fa4baa10a3644c6cd78878259ba3be3417db7a0ef9f37b4e72f4c5
+oid sha256:c9dd27b602a45f99923c24da8f9e5befe79844b54bd93cb779da619a459cd922
 size 49615

training_history.json

@@ -1,128 +1,95 @@
 {
   "val_auc": [
-    0.7467547568052559,
-    0.756945153471315,
-    0.7634001469039655,
-    0.7617216066140435,
-    0.772287560248718,
-    0.7758325921681795,
-    0.7762064205837024,
-    0.7759986485019552,
-    0.7717502107106219,
-    0.7817622206975103,
-    0.7819540900908044,
-    0.7747707996217466,
-    0.7855751783463463,
-    0.7844577263903272,
-    0.7895011833674472,
-    0.7870999926478874,
-    0.7866294804960441,
-    0.7857090835925971,
-    0.7843587379165495,
-    0.7846049991138284,
-    0.7851299030696568,
-    0.7874004151754055,
-    0.7883833857008096,
-    0.7869940899617391,
-    0.7856879952448657,
-    0.7839751143668345,
-    0.7838808506048438,
-    0.7847943332959197,
-    0.7856171245680633,
-    0.7864544356861861,
-    0.7853772013987912,
-    0.7859503896043434,
-    0.7866153063606836,
-    0.7865681744796882,
-    0.7871663690866486,
-    0.786776292272054,
-    0.7868910681974118,
-    0.7871056392546572,
-    0.7871857288812873,
-    0.7870264715230098
+    0.7736832008136094,
+    0.7747291050295857,
+    0.7721354937130178,
+    0.7754521079881656,
+    0.771109236316568,
+    0.7707953494822486,
+    0.7769595968934911,
+    0.7790236686390533,
+    0.7828982525887573,
+    0.7795282451923076,
+    0.7766450166420119,
+    0.7781062777366865,
+    0.7773546135355028,
+    0.7789327154215976,
+    0.7784307969674555,
+    0.7786693324704143,
+    0.7788579419378697,
+    0.7791343842455619,
+    0.7784714774408283,
+    0.7804860854289941,
+    0.7772855029585798,
+    0.7775802052514793,
+    0.7786233357988165,
+    0.778695451183432,
+    0.7788134476701183,
+    0.7789931582840236,
+    0.7788956176035503,
+    0.7789702755177513,
+    0.7785478688979289
   ],
   "val_bal_acc": [
-    0.6980316850696111,
-    0.685309649543927,
-    0.7097935669707154,
-    0.716034565530485,
-    0.7167546807489198,
-    0.7251560249639942,
-    0.7309169467114738,
-    0.7186749879980796,
-    0.714594335093615,
-    0.7345175228036486,
-    0.7385981757081133,
-    0.7393182909265483,
-    0.7390782525204033,
-    0.7390782525204033,
-    0.742198751800288,
-    0.7385981757081133,
-    0.7412385981757081,
-    0.7349975996159386,
-    0.7378780604896784,
-    0.7450792126740279,
-    0.7450792126740279,
-    0.7443590974555929,
-    0.7400384061449832,
-    0.7395583293326933,
-    0.7383581373019683,
-    0.7321171387421987,
-    0.7345175228036486,
-    0.7335573691790687,
-    0.7347575612097936,
-    0.7393182909265483,
-    0.7328372539606337,
-    0.7342774843975035,
-    0.7347575612097936,
-    0.7371579452712433,
-    0.7378780604896784,
-    0.7373979836773883,
-    0.7376380220835335,
-    0.7385981757081133,
-    0.7378780604896784,
-    0.7378780604896784
+    0.7055288461538461,
+    0.7146634615384615,
+    0.7165865384615384,
+    0.7314903846153846,
+    0.7286057692307693,
+    0.7264423076923077,
+    0.7286057692307693,
+    0.726201923076923,
+    0.7358173076923077,
+    0.7336538461538462,
+    0.7076923076923077,
+    0.7276442307692308,
+    0.7276442307692308,
+    0.7254807692307692,
+    0.7290865384615385,
+    0.7278846153846155,
+    0.7266826923076923,
+    0.7295673076923077,
+    0.7298076923076924,
+    0.7317307692307693,
+    0.7314903846153846,
+    0.7324519230769231,
+    0.7310096153846154,
+    0.7307692307692308,
+    0.7326923076923078,
+    0.7329326923076923,
+    0.7298076923076924,
+    0.73125,
+    0.7302884615384615
   ],
   "train_loss": [
-    8.771085142482116,
-    4.2970731621687195,
-    2.68378420419523,
-    1.9316510652746937,
-    1.4755360628328016,
-    1.1796510363272779,
-    1.0143797920545616,
-    0.8811894825996487,
-    0.7747690477274067,
-    0.7082606209710822,
-    0.6556297140433145,
-    0.6177907896517493,
-    0.5899819823042038,
-    0.5435396304610225,
-    0.5319194863020665,
-    0.5017238825003036,
-    0.48398977930456755,
-    0.46635610672229416,
-    0.45954567178805533,
-    0.43131048228522106,
-    0.4212635166576236,
-    0.4121998123483059,
-    0.4041198058929832,
-    0.39608139505511836,
-    0.3879514780584944,
-    0.37410734596398165,
-    0.37554909878351894,
-    0.3683332845391563,
-    0.36570364750403095,
-    0.3616273019380602,
-    0.3575036424656473,
-    0.355650919567041,
-    0.34983081677976,
-    0.3488211069123247,
-    0.3464294295055959,
-    0.34449730044411075,
-    0.3416846572644439,
-    0.34562161873475805,
-    0.3453929291718682,
-    0.34312185909478893
+    0.6120673337498227,
+    0.5313478146453161,
+    0.4957638501718238,
+    0.4757815355787406,
+    0.46550961020025045,
+    0.45031767560018077,
+    0.44026057140247243,
+    0.42258331622626333,
+    0.39576308388967774,
+    0.3870376779420956,
+    0.3789778110143301,
+    0.3785893202230737,
+    0.3678017641241486,
+    0.3540323513585168,
+    0.34848260839243195,
+    0.3459027508223379,
+    0.34450753123776334,
+    0.33642298667817505,
+    0.3345084897166974,
+    0.32999936602002866,
+    0.32982844557311086,
+    0.32826521489265803,
+    0.32426364115766576,
+    0.3250381461269147,
+    0.32325289146722974,
+    0.3226573482357167,
+    0.31961060899334987,
+    0.32056339558314634,
+    0.3206416093618483
   ]
 }