| { |
| "checkpoint": { |
| "path": "models/pimt_v6.pt", |
| "epoch": 2, |
| "val_loss": 0.484091709347537, |
| "hidden_dim": 256, |
| "embedding_dim": 151, |
| "state_dim": 2, |
| "num_layers": 4, |
| "num_heads": 4 |
| }, |
| "data": { |
| "path": "data/empirical_dataset_v8.jsonl", |
| "n_records": 6078, |
| "n_validation_samples": 1089 |
| }, |
| "prediction": { |
| "profile_mse": 0.01670854724943638, |
| "profile_mae": 0.08257085829973221, |
| "profile_prediction_variance": 0.003912549465894699, |
| "tier_prediction_variance": 0.009843787178397179, |
| "tier_metrics": { |
| "top": { |
| "mse": 0.03479713946580887, |
| "mae": 0.06280406564474106, |
| "prediction_variance": 0.0011337018804624677, |
| "target_variance": 0.03625021129846573, |
| "mean_pairwise_cosine_similarity": 0.9688117237632773, |
| "mean_spearman_rho": 0.21981701217589275 |
| }, |
| "middle": { |
| "mse": 0.020692989230155945, |
| "mae": 0.0404493473470211, |
| "prediction_variance": 0.0006535974680446088, |
| "target_variance": 0.02135564014315605, |
| "mean_pairwise_cosine_similarity": 0.9567461467957046, |
| "mean_spearman_rho": 0.20025567720380308 |
| }, |
| "base": { |
| "mse": 0.09510858356952667, |
| "mae": 0.1818365752696991, |
| "prediction_variance": 0.021325431764125824, |
| "target_variance": 0.11628847569227219, |
| "mean_pairwise_cosine_similarity": 0.9431027251998285, |
| "mean_spearman_rho": 0.32991390136684196 |
| } |
| } |
| }, |
| "spearman": { |
| "flat_spearman_rho": 0.27788543250229925, |
| "flat_spearman_p_value": 0.0, |
| "profile_spearman_rho": 0.3537687699618818, |
| "profile_spearman_p_value": 0.0 |
| }, |
| "top_descriptors": { |
| "top": { |
| "predicted": [ |
| { |
| "descriptor": "fresh", |
| "score": 0.1774865686893463 |
| }, |
| { |
| "descriptor": "sulfurous", |
| "score": 0.1653631627559662 |
| }, |
| { |
| "descriptor": "fishy", |
| "score": 0.13863913714885712 |
| }, |
| { |
| "descriptor": "fatty", |
| "score": 0.1347414255142212 |
| }, |
| { |
| "descriptor": "earthy", |
| "score": 0.130441352725029 |
| } |
| ], |
| "target": [ |
| { |
| "descriptor": "fresh", |
| "score": 0.2029384821653366 |
| }, |
| { |
| "descriptor": "sulfurous", |
| "score": 0.18824610114097595 |
| }, |
| { |
| "descriptor": "fishy", |
| "score": 0.15518824756145477 |
| }, |
| { |
| "descriptor": "earthy", |
| "score": 0.1515151560306549 |
| }, |
| { |
| "descriptor": "popcorn", |
| "score": 0.127640038728714 |
| } |
| ] |
| }, |
| "middle": { |
| "predicted": [ |
| { |
| "descriptor": "fresh", |
| "score": 0.15246763825416565 |
| }, |
| { |
| "descriptor": "sulfurous", |
| "score": 0.1356164962053299 |
| }, |
| { |
| "descriptor": "grassy", |
| "score": 0.09715040773153305 |
| }, |
| { |
| "descriptor": "fishy", |
| "score": 0.07722806930541992 |
| }, |
| { |
| "descriptor": "earthy", |
| "score": 0.0631425753235817 |
| } |
| ], |
| "target": [ |
| { |
| "descriptor": "fresh", |
| "score": 0.1873278170824051 |
| }, |
| { |
| "descriptor": "sulfurous", |
| "score": 0.15059687197208405 |
| }, |
| { |
| "descriptor": "earthy", |
| "score": 0.09182736277580261 |
| }, |
| { |
| "descriptor": "grassy", |
| "score": 0.09182736277580261 |
| }, |
| { |
| "descriptor": "fishy", |
| "score": 0.07346189022064209 |
| } |
| ] |
| }, |
| "base": { |
| "predicted": [ |
| { |
| "descriptor": "sulfurous", |
| "score": 0.8006615042686462 |
| }, |
| { |
| "descriptor": "fresh", |
| "score": 0.663232147693634 |
| }, |
| { |
| "descriptor": "fermented", |
| "score": 0.5910075306892395 |
| }, |
| { |
| "descriptor": "waxy", |
| "score": 0.5055632591247559 |
| }, |
| { |
| "descriptor": "grassy", |
| "score": 0.4854605197906494 |
| } |
| ], |
| "target": [ |
| { |
| "descriptor": "sulfurous", |
| "score": 0.8126721978187561 |
| }, |
| { |
| "descriptor": "fermented", |
| "score": 0.6391184329986572 |
| }, |
| { |
| "descriptor": "fresh", |
| "score": 0.6391184329986572 |
| }, |
| { |
| "descriptor": "waxy", |
| "score": 0.5325987339019775 |
| }, |
| { |
| "descriptor": "balsamic", |
| "score": 0.5307621955871582 |
| } |
| ] |
| } |
| } |
| } |
