| { | |
| "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 | |
| } | |
| } |