metrics_v8.json

{
  "spearman": {
    "spearman_rho": -0.051573076971065016,
    "spearman_p_value": 0.0,
    "n_pairs": 18468003
  },
  "retrieval_mAP": {
    "mAP": 0.2616184066751507,
    "AP_std": 0.12348274255924709,
    "n_queries": 6078
  },
  "prediction": {
    "profile_mse": 0.015763238072395325,
    "profile_mae": 0.07920028269290924,
    "prediction_variance": 0.003771901363506913,
    "tier_prediction_variance": 0.009696539491415024
  },
  "n_samples": 6078,
  "embedding_dim": 138,
  "checkpoint": {
    "epoch": 4,
    "val_loss": 0.48002281219419773,
    "hidden_dim": 256,
    "embedding_dim": 151,
    "state_dim": 2,
    "num_layers": 4,
    "num_heads": 4
  }
}