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