1{
 2  "device": "cuda",
 3  "math_check": {
 4    "posterior_sum": 1.0,
 5    "marginal_max_error": 0.0,
 6    "pair_mass_before": 0.09989038109779358,
 7    "pair_mass_after_boost": 0.4505513608455658
 8  },
 9  "baseline": {
10    "f1": 0.9498525261878967,
11    "reconstruction_mse": 0.1092461496591568
12  },
13  "subset_attention": {
14    "f1": 0.9874153137207031,
15    "reconstruction_mse": 0.0055355895310640335
16  },
17  "subsets_per_neighborhood": 63,
18  "steps": 500,
19  "seed": 283
20}