Upload CAUM (jax) trained on MIND-small — 3 seeds

seed_42/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:46661584d959a93125cf9e06b08b854139a5e1b41f9fcaf59f431ddde61d5553
+oid sha256:0fdb3f797c03b0806b9d95206884cb6473b2633417d76fa09c756f15edfefa39
 size 47322396

seed_42/test_results.json

@@ -1,8 +1,8 @@
 {
-  "loss": 4.885279593137135,
-  "auc": 0.6716255651925904,
-  "mrr": 0.31951080672544796,
-  "ndcg@5": 0.3533953887559558,
-  "ndcg@10": 0.417788739745737,
+  "loss": 4.885290258142447,
+  "auc": 0.6716320262703661,
+  "mrr": 0.31951266628908737,
+  "ndcg@5": 0.35341487453541737,
+  "ndcg@10": 0.4177790125019271,
   "num_impressions": 72903.0
 }

seed_42/training_run_summary.json

@@ -257,44 +257,44 @@
     "initial_validation_metrics": {},
     "best_validation_summary": {
         "epoch_number": 8.0,
-        "train_loss": 1.2569683923193469,
-        "average_metric_value": 0.5173871047050278,
-        "val_loss": 4.491243967664317,
-        "val_auc": 0.7467889007717999,
-        "val_mrr": 0.3941287371779766,
-        "val_ndcg@5": 0.43629077698141816,
-        "val_ndcg@10": 0.4923400038889164,
+        "train_loss": 1.256968167472918,
+        "average_metric_value": 0.5174154410910897,
+        "val_loss": 4.49124723870549,
+        "val_auc": 0.7468131525042653,
+        "val_mrr": 0.3941594308823248,
+        "val_ndcg@5": 0.4363602148970487,
+        "val_ndcg@10": 0.49232896608071997,
         "val_num_impressions": 7824.0,
         "timing": {
             "epoch_training_times": [
-                163.71576189994812,
-                128.66470193862915,
-                124.40946054458618,
-                122.63417339324951,
-                123.02388978004456,
-                124.40860843658447,
-                123.64945530891418,
-                124.03472113609314
+                163.05336737632751,
+                125.91719460487366,
+                123.6555745601654,
+                124.48746538162231,
+                126.33813762664795,
+                125.54173946380615,
+                122.81819033622742,
+                124.80581545829773
             ],
             "epoch_validation_times": [
-                225.55031847953796,
-                155.1456437110901,
-                154.2360875606537,
-                154.55679368972778,
-                155.49029302597046,
-                155.03285694122314,
-                156.5079698562622,
-                172.0624237060547
+                224.672523021698,
+                151.31402707099915,
+                151.30706453323364,
+                151.26560187339783,
+                159.7723400592804,
+                153.1495270729065,
+                153.66180729866028,
+                155.1824631690979
             ],
-            "total_training_time": 2363.861449956894
+            "total_training_time": 2337.687963247299
         }
     },
     "final_test_metrics": {
-        "loss": 4.885279593137135,
-        "auc": 0.6716255651925904,
-        "mrr": 0.31951080672544796,
-        "ndcg@5": 0.3533953887559558,
-        "ndcg@10": 0.417788739745737,
+        "loss": 4.885290258142447,
+        "auc": 0.6716320262703661,
+        "mrr": 0.31951266628908737,
+        "ndcg@5": 0.35341487453541737,
+        "ndcg@10": 0.4177790125019271,
         "num_impressions": 72903.0
     }
 }