Upload trained model (Safetensors) Acc: 72.79%

README.md

@@ -4,7 +4,8 @@ tags:
 - fairsteer
 - bias-detection
 - tinyllama
-library_name: pytorch
+- safetensors
+library_name: safetensors
 pipeline_tag: text-classification
 ---
 
@@ -13,11 +14,11 @@ pipeline_tag: text-classification
 Biased Activation Detection (BAD) classifier for TinyLlama-1.1B.
 
 ## Artifacts
-- **Model**: `pytorch_model.bin`
+- **Model**: `model.safetensors` (SafeTensors format)
 - **Scaler**: `scaler.pkl` (StandardScaler)
 - **Config**: `config.json`
 
 ## Stats
-- **Balanced Accuracy**: 76.47%
-- **Best Layer**: 11
-- **Training Date**: 2025-11-22
+- **Balanced Accuracy**: 72.79%
+- **Best Layer**: 20
+- **Training Date**: 2025-12-12

config.json

@@ -1,8 +1,8 @@
 {
   "input_dim": 2048,
-  "layer_idx": 11,
+  "layer_idx": 20,
   "base_model": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
-  "best_val_bal_acc": 0.7647058823529411,
+  "best_val_bal_acc": 0.7279477947794779,
   "training_method": "GroupSplit + Standardized + Balanced",
-  "training_date": "2025-11-22T06:49:56.551085"
+  "training_date": "2025-12-12T05:40:01.622551"
 }

model.safetensors

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

scaler.pkl

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

training_history.json

@@ -1,62 +1,98 @@
 {
   "val_auc": [
-    0.6721453287197232,
-    0.7370242214532872,
-    0.7352941176470589,
-    0.7448096885813149,
-    0.7543252595155708,
-    0.7569204152249136,
-    0.7577854671280276,
-    0.759515570934256,
-    0.7655709342560553,
-    0.7698961937716262,
-    0.7707612456747406,
-    0.7716262975778547,
-    0.7707612456747405,
-    0.7690311418685121,
-    0.7698961937716263,
-    0.7690311418685121,
-    0.7690311418685121,
-    0.7681660899653979
+    0.7273940490358668,
+    0.7504346519260386,
+    0.7575085816412425,
+    0.7573322701307035,
+    0.7596420420619919,
+    0.7644952163983275,
+    0.7669964903781107,
+    0.7635729482039112,
+    0.7664611758205524,
+    0.7653390001466394,
+    0.7669162843377048,
+    0.7698281079233036,
+    0.7678008799979907,
+    0.7703854993960242,
+    0.7706289530843273,
+    0.7703307121891307,
+    0.7706585239982144,
+    0.7678606294472832,
+    0.7682736095391718,
+    0.7705408479641844,
+    0.7703784104783117,
+    0.7698073475214309,
+    0.767579604495103,
+    0.7694949287908088,
+    0.7691457489583342,
+    0.7687167681664656,
+    0.7699388975801271,
+    0.7697089128354779,
+    0.7699554046313721,
+    0.7698038030625745
   ],
   "val_bal_acc": [
-    0.6323529411764706,
-    0.7058823529411764,
-    0.7647058823529411,
-    0.7647058823529411,
-    0.7352941176470589,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176,
-    0.7205882352941176
+    0.6671917191719172,
+    0.6874437443744374,
+    0.7031953195319531,
+    0.7029702970297029,
+    0.7088208820882088,
+    0.7200720072007201,
+    0.711971197119712,
+    0.711971197119712,
+    0.716021602160216,
+    0.7133213321332134,
+    0.7223222322232223,
+    0.7236723672367237,
+    0.7209720972097209,
+    0.725922592259226,
+    0.7279477947794779,
+    0.7232223222322232,
+    0.7214221422142215,
+    0.718046804680468,
+    0.7178217821782178,
+    0.7205220522052205,
+    0.7225472547254725,
+    0.7184968496849685,
+    0.7198469846984699,
+    0.7227722772277227,
+    0.7193969396939695,
+    0.7252475247524752,
+    0.7216471647164716,
+    0.7227722772277227,
+    0.7214221422142214,
+    0.7241224122412241
   ],
   "train_loss": [
-    0.7645134925842285,
-    0.6531355579694113,
-    0.5934327244758606,
-    0.5382237235705057,
-    0.5212106804052988,
-    0.5007542173067728,
-    0.49714147051175434,
-    0.48595739404360455,
-    0.48224801818529767,
-    0.4633047580718994,
-    0.4640958805878957,
-    0.4591399331887563,
-    0.4668729603290558,
-    0.45914029081662494,
-    0.4610348045825958,
-    0.46088655789693195,
-    0.45236412684122723,
-    0.46567730108896893
+    0.6589798988967106,
+    0.5834655373260893,
+    0.5626362512851584,
+    0.5517601516740075,
+    0.5468299312838193,
+    0.5445838414389511,
+    0.5378933320785391,
+    0.5344717323780059,
+    0.5338149806548809,
+    0.5299872453870444,
+    0.5244478642940521,
+    0.5223940512229657,
+    0.5225685875991295,
+    0.5207774322608422,
+    0.5200736000620085,
+    0.5207284010689834,
+    0.5184559558999949,
+    0.517919123994893,
+    0.518367536725669,
+    0.5166512495484845,
+    0.5150178672938511,
+    0.5143926667756048,
+    0.5142317825350268,
+    0.5123990134946231,
+    0.5112694777291397,
+    0.513259904754573,
+    0.512855320757833,
+    0.5109211087226868,
+    0.510224807467954,
+    0.5088399624002391
   ]
 }