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