{
  "spearman": {
    "spearman_rho": -0.05881833791439946,
    "spearman_p_value": 0.0,
    "n_pairs": 18468003
  },
  "retrieval_mAP": {
    "mAP": 0.25191517122784174,
    "AP_std": 0.11215366222930846,
    "n_queries": 6078
  },
  "prediction": {
    "profile_mse": 0.01602626033127308,
    "profile_mae": 0.08081231266260147,
    "prediction_variance": 0.003824461717158556,
    "tier_prediction_variance": 0.009652215987443924
  },
  "n_samples": 6078,
  "embedding_dim": 138,
  "checkpoint": {
    "epoch": 2,
    "val_loss": 0.484091709347537,
    "hidden_dim": 256,
    "embedding_dim": 151,
    "state_dim": 2,
    "num_layers": 4,
    "num_heads": 4
  }
}