1{
 2  "baseline": {
 3    "val_mse": 0.02755516953766346,
 4    "pairwise_distance": 0.7544022798538208,
 5    "pk_nll": 0.18817760050296783,
 6    "last_train_task": 0.025835629552602768
 7  },
 8  "poisson_kernel": {
 9    "val_mse": 0.027630342170596123,
10    "pairwise_distance": 0.7214926481246948,
11    "pk_nll": -0.039105549454689026,
12    "last_train_task": 0.026616107672452927
13  }
14}