training_stats.json

{
  "best_val_action_acc": 0.6343636363636364,
  "epochs": 30,
  "train_samples": 99000,
  "val_samples": 11000,
  "collected_samples": 20900,
  "source_counts": {
    "gemini": 350,
    "groq": 500,
    "nn": 20050
  },
  "model_size_kb": 5563.896484375,
  "timestamp": "2026-03-10T18:51:06",
  "history": {
    "train_loss": [
      2.1586153556391134,
      2.022064336181916,
      1.9399496273896129,
      1.8850172924012254,
      1.8444405935474277,
      1.8108722862509108,
      1.7853464920496203,
      1.7663688782564144,
      1.7439076089367425,
      1.7264809012413025,
      1.7118696994388227,
      1.6935583586545335,
      1.6788606901758725,
      1.6657644467255504,
      1.6530226524343197,
      1.641498893192134,
      1.6282407420197713,
      1.6164132405802147,
      1.603588941785478,
      1.595070652126037,
      1.5846694302313107,
      1.5778400799662797,
      1.5662498191459893,
      1.5585865790082007,
      1.5538890951687527,
      1.546544762616305,
      1.5440150818873926,
      1.538954149816454,
      1.5357574169168766,
      1.5373721251782684
    ],
    "val_loss": [
      2.0330969745462593,
      1.950374180620367,
      1.8876040632074529,
      1.8691241632808338,
      1.8585692102258855,
      1.8160562298514626,
      1.7953031604940242,
      1.7810519608584316,
      1.7864647778597744,
      1.7562655427239158,
      1.7717987515709617,
      1.7592156691984697,
      1.7528499581597068,
      1.755829854445024,
      1.7492482228712602,
      1.7700981226834385,
      1.7627330259843306,
      1.766843156381087,
      1.7668060822920366,
      1.7652146057649092,
      1.7598941759629683,
      1.7872037345712835,
      1.796427683396773,
      1.7877931811592795,
      1.795323371887207,
      1.8082295331087979,
      1.7898864637721668,
      1.7959014827554876,
      1.8005776405334473,
      1.7997321974147449
    ],
    "val_action_acc": [
      0.6153636363636363,
      0.6068181818181818,
      0.619,
      0.6070909090909091,
      0.6047272727272728,
      0.6151818181818182,
      0.6208181818181818,
      0.6241818181818182,
      0.6190909090909091,
      0.6285454545454545,
      0.6245454545454545,
      0.6277272727272727,
      0.6298181818181818,
      0.626,
      0.6315454545454545,
      0.622,
      0.6302727272727273,
      0.624,
      0.6313636363636363,
      0.6272727272727273,
      0.6263636363636363,
      0.6334545454545455,
      0.6288181818181818,
      0.6343636363636364,
      0.6314545454545455,
      0.6289090909090909,
      0.6302727272727273,
      0.6316363636363637,
      0.6307272727272727,
      0.6314545454545455
    ],
    "val_loc_acc": [
      0.5850909090909091,
      0.5952727272727273,
      0.6050909090909091,
      0.6046363636363636,
      0.6033636363636363,
      0.617909090909091,
      0.6194545454545455,
      0.6222727272727273,
      0.624,
      0.6235454545454545,
      0.6247272727272727,
      0.6235454545454545,
      0.6262727272727273,
      0.6247272727272727,
      0.6276363636363637,
      0.6249090909090909,
      0.6272727272727273,
      0.6284545454545455,
      0.6277272727272727,
      0.6288181818181818,
      0.6287272727272727,
      0.6265454545454545,
      0.6269090909090909,
      0.6293636363636363,
      0.6292727272727273,
      0.6283636363636363,
      0.6292727272727273,
      0.629,
      0.6291818181818182,
      0.6298181818181818
    ]
  }
}