{
  "device": "cuda",
  "math_check": {
    "posterior_sum": 1.0,
    "marginal_max_error": 0.0,
    "pair_mass_before": 0.09989038109779358,
    "pair_mass_after_boost": 0.4505513608455658
  },
  "baseline": {
    "f1": 0.9498525261878967,
    "reconstruction_mse": 0.1092461496591568
  },
  "subset_attention": {
    "f1": 0.9874153137207031,
    "reconstruction_mse": 0.0055355895310640335
  },
  "subsets_per_neighborhood": 63,
  "steps": 500,
  "seed": 283
}