| { | |
| "model_name": "intfloat/multilingual-e5-small", | |
| "pooling": "mean", | |
| "text_prefix": "query: ", | |
| "max_len": 128, | |
| "function_labels": [ | |
| "planning", | |
| "receipt", | |
| "appreciation", | |
| "affiliation", | |
| "deliberation", | |
| "brisk", | |
| "transition" | |
| ], | |
| "tone_labels": [ | |
| "thoughtful", | |
| "attentive", | |
| "gentle", | |
| "pleased", | |
| "brisk", | |
| "reflective" | |
| ], | |
| "tone_collapse": { | |
| "curious": "thoughtful", | |
| "calm": "attentive", | |
| "focused": "brisk", | |
| "concerned": "gentle" | |
| }, | |
| "num_functions": 7, | |
| "num_tones": 6, | |
| "dropout": 0.15, | |
| "contrastive_lambda": 0.3, | |
| "training_info": { | |
| "lr": 3e-05, | |
| "epochs_run": 7, | |
| "best_val_loss": 1.5853182556806669, | |
| "test_func_acc": 0.7118353344768439, | |
| "test_tone_acc": 0.6826758147512865, | |
| "cpu_latency_ms_colab": 80.43561199997384, | |
| "params": 117658765, | |
| "train_rows": 9965, | |
| "val_rows": 1303, | |
| "test_rows": 1166, | |
| "label_files": [ | |
| "spring_ml.jsonl", | |
| "synthetic_targeted_v1.jsonl", | |
| "synthetic_targeted_v2.jsonl", | |
| "synthetic_targeted_v3.jsonl", | |
| "synthetic_targeted_v4.jsonl" | |
| ], | |
| "variant": "contrastive_candidates" | |
| } | |
| } |