{
  "train": [
    {
      "loss": 20.609816724603828
    },
    {
      "loss": 17.001067876815796
    },
    {
      "loss": 15.968297524885697
    },
    {
      "loss": 15.571664810180664
    },
    {
      "loss": 15.281125415455211
    },
    {
      "loss": 15.190964070233433
    },
    {
      "loss": 15.045507322658192
    },
    {
      "loss": 14.99332122369246
    },
    {
      "loss": 15.03947028246793
    },
    {
      "loss": 14.927926605398005
    },
    {
      "loss": 14.904001929543234
    },
    {
      "loss": 14.852870551022617
    },
    {
      "loss": 14.837879831140691
    },
    {
      "loss": 14.785482840104537
    },
    {
      "loss": 14.830404389988292
    },
    {
      "loss": 14.738714521581477
    },
    {
      "loss": 14.788421045650136
    },
    {
      "loss": 14.77198548750444
    },
    {
      "loss": 14.674884275956588
    },
    {
      "loss": 14.686285083944147
    },
    {
      "loss": 14.662966077977961
    },
    {
      "loss": 14.692009145563299
    },
    {
      "loss": 14.642124024304477
    },
    {
      "loss": 14.673041690479625
    },
    {
      "loss": 14.604451547969472
    },
    {
      "loss": 14.592413295399059
    },
    {
      "loss": 14.584426338022405
    },
    {
      "loss": 14.664776000109585
    },
    {
      "loss": 14.555745514956387
    },
    {
      "loss": 14.563685395500876
    },
    {
      "loss": 14.56706645271995
    },
    {
      "loss": 14.4845242283561
    },
    {
      "loss": 14.557521018114956
    },
    {
      "loss": 14.563952380960638
    },
    {
      "loss": 14.541412050073797
    },
    {
      "loss": 14.50578971342607
    },
    {
      "loss": 14.445949576117776
    },
    {
      "loss": 14.498417290774258
    },
    {
      "loss": 14.475244392048229
    },
    {
      "loss": 14.44333902272311
    },
    {
      "loss": 14.46074891090393
    },
    {
      "loss": 14.419529828158291
    },
    {
      "loss": 14.437433177774603
    },
    {
      "loss": 14.499378442764282
    },
    {
      "loss": 14.431849132884633
    },
    {
      "loss": 14.424177798357876
    },
    {
      "loss": 14.39670426195318
    },
    {
      "loss": 14.348045327446677
    },
    {
      "loss": 14.464565645564686
    },
    {
      "loss": 14.438084710728038
    },
    {
      "loss": 14.310053673657505
    },
    {
      "loss": 14.380919088016856
    },
    {
      "loss": 14.357033447785811
    },
    {
      "loss": 14.369577277790416
    },
    {
      "loss": 14.351079377261074
    }
  ],
  "val": [
    {
      "loss": 18.246644496917725,
      "mse": 18.257070541381836,
      "mae": 2.767958402633667,
      "r2": 0.27587616443634033,
      "r2_mean": 0.01685653068125248,
      "r2_median": 0.03934532403945923,
      "r2_min": -33.37522888183594,
      "r2_max": 0.35776031017303467,
      "cosine_similarity": 0.5018709897994995
    },
    {
      "loss": 16.364336585998537,
      "mse": 16.373306274414062,
      "mae": 2.6759896278381348,
      "r2": 0.350591242313385,
      "r2_mean": 0.09112238883972168,
      "r2_median": 0.06944680213928223,
      "r2_min": -22.05408477783203,
      "r2_max": 0.3918255567550659,
      "cosine_similarity": 0.5534306168556213
    },
    {
      "loss": 15.788817596435546,
      "mse": 15.799945831298828,
      "mae": 2.62270450592041,
      "r2": 0.3733322024345398,
      "r2_mean": 0.12282542884349823,
      "r2_median": 0.08083301782608032,
      "r2_min": -15.787790298461914,
      "r2_max": 0.41044384241104126,
      "cosine_similarity": 0.5725873708724976
    },
    {
      "loss": 15.497636318206787,
      "mse": 15.508538246154785,
      "mae": 2.5950748920440674,
      "r2": 0.38489025831222534,
      "r2_mean": 0.13708166778087616,
      "r2_median": 0.08684241771697998,
      "r2_min": -11.413467407226562,
      "r2_max": 0.4362717866897583,
      "cosine_similarity": 0.5857664346694946
    },
    {
      "loss": 15.351541709899902,
      "mse": 15.363690376281738,
      "mae": 2.568354845046997,
      "r2": 0.39063531160354614,
      "r2_mean": 0.13934743404388428,
      "r2_median": 0.08826941251754761,
      "r2_min": -8.184919357299805,
      "r2_max": 0.4567607045173645,
      "cosine_similarity": 0.5936097502708435
    },
    {
      "loss": 15.255533027648926,
      "mse": 15.268318176269531,
      "mae": 2.5600216388702393,
      "r2": 0.39441800117492676,
      "r2_mean": 0.14353397488594055,
      "r2_median": 0.09055894613265991,
      "r2_min": -6.149453639984131,
      "r2_max": 0.45417433977127075,
      "cosine_similarity": 0.598899245262146
    },
    {
      "loss": 15.156796932220459,
      "mse": 15.170814514160156,
      "mae": 2.5517032146453857,
      "r2": 0.39828526973724365,
      "r2_mean": 0.14761891961097717,
      "r2_median": 0.09417349100112915,
      "r2_min": -4.62712287902832,
      "r2_max": 0.45061182975769043,
      "cosine_similarity": 0.6018620133399963
    },
    {
      "loss": 15.090502452850341,
      "mse": 15.105701446533203,
      "mae": 2.5358567237854004,
      "r2": 0.4008678197860718,
      "r2_mean": 0.1517454981803894,
      "r2_median": 0.09789353609085083,
      "r2_min": -4.215254306793213,
      "r2_max": 0.4557972550392151,
      "cosine_similarity": 0.6066699028015137
    },
    {
      "loss": 15.056775283813476,
      "mse": 15.071660041809082,
      "mae": 2.5236780643463135,
      "r2": 0.4022179841995239,
      "r2_mean": 0.15281586349010468,
      "r2_median": 0.09995847940444946,
      "r2_min": -3.621105194091797,
      "r2_max": 0.46177375316619873,
      "cosine_similarity": 0.6086801290512085
    },
    {
      "loss": 15.04591760635376,
      "mse": 15.062102317810059,
      "mae": 2.5166077613830566,
      "r2": 0.4025970697402954,
      "r2_mean": 0.14819204807281494,
      "r2_median": 0.09973281621932983,
      "r2_min": -3.0591392517089844,
      "r2_max": 0.46009916067123413,
      "cosine_similarity": 0.6097062826156616
    },
    {
      "loss": 15.013768100738526,
      "mse": 15.029498100280762,
      "mae": 2.5237395763397217,
      "r2": 0.4038902521133423,
      "r2_mean": 0.15404430031776428,
      "r2_median": 0.10138571262359619,
      "r2_min": -2.6930553913116455,
      "r2_max": 0.46185535192489624,
      "cosine_similarity": 0.6109820008277893
    },
    {
      "loss": 14.972628974914551,
      "mse": 14.991690635681152,
      "mae": 2.519646406173706,
      "r2": 0.40538978576660156,
      "r2_mean": 0.15686893463134766,
      "r2_median": 0.10368794202804565,
      "r2_min": -2.356921434402466,
      "r2_max": 0.4682624340057373,
      "cosine_similarity": 0.6120056509971619
    },
    {
      "loss": 14.983994579315185,
      "mse": 15.000967979431152,
      "mae": 2.507404088973999,
      "r2": 0.4050217866897583,
      "r2_mean": 0.1516677439212799,
      "r2_median": 0.10245764255523682,
      "r2_min": -2.0743730068206787,
      "r2_max": 0.46684253215789795,
      "cosine_similarity": 0.6127941608428955
    },
    {
      "loss": 14.923584842681885,
      "mse": 14.940834999084473,
      "mae": 2.510566234588623,
      "r2": 0.40740686655044556,
      "r2_mean": 0.15889577567577362,
      "r2_median": 0.10642272233963013,
      "r2_min": -1.639467477798462,
      "r2_max": 0.4676855802536011,
      "cosine_similarity": 0.6133753061294556
    },
    {
      "loss": 14.885868167877197,
      "mse": 14.90212345123291,
      "mae": 2.5070483684539795,
      "r2": 0.40894222259521484,
      "r2_mean": 0.16219769418239594,
      "r2_median": 0.10877114534378052,
      "r2_min": -1.4757106304168701,
      "r2_max": 0.4795358180999756,
      "cosine_similarity": 0.6141579747200012
    },
    {
      "loss": 14.902330780029297,
      "mse": 14.920063972473145,
      "mae": 2.5075578689575195,
      "r2": 0.408230721950531,
      "r2_mean": 0.15961791574954987,
      "r2_median": 0.10700821876525879,
      "r2_min": -1.505033254623413,
      "r2_max": 0.4780617356300354,
      "cosine_similarity": 0.6145411133766174
    },
    {
      "loss": 14.880312633514404,
      "mse": 14.897867202758789,
      "mae": 2.4977564811706543,
      "r2": 0.4091110825538635,
      "r2_mean": 0.16172455251216888,
      "r2_median": 0.1085314154624939,
      "r2_min": -1.1947932243347168,
      "r2_max": 0.4761975407600403,
      "cosine_similarity": 0.6147447228431702
    },
    {
      "loss": 14.896513748168946,
      "mse": 14.91469955444336,
      "mae": 2.5020458698272705,
      "r2": 0.4084434509277344,
      "r2_mean": 0.15899036824703217,
      "r2_median": 0.10672199726104736,
      "r2_min": -1.2995786666870117,
      "r2_max": 0.47241640090942383,
      "cosine_similarity": 0.6148735284805298
    },
    {
      "loss": 14.889157772064209,
      "mse": 14.909862518310547,
      "mae": 2.508105754852295,
      "r2": 0.40863531827926636,
      "r2_mean": 0.15980710089206696,
      "r2_median": 0.10755527019500732,
      "r2_min": -1.0859878063201904,
      "r2_max": 0.46420425176620483,
      "cosine_similarity": 0.6146147847175598
    },
    {
      "loss": 14.892903995513915,
      "mse": 14.90970516204834,
      "mae": 2.4925057888031006,
      "r2": 0.408641517162323,
      "r2_mean": 0.15805493295192719,
      "r2_median": 0.10711073875427246,
      "r2_min": -1.0736799240112305,
      "r2_max": 0.47862738370895386,
      "cosine_similarity": 0.6149945855140686
    },
    {
      "loss": 14.898202037811279,
      "mse": 14.91330623626709,
      "mae": 2.4915273189544678,
      "r2": 0.40849870443344116,
      "r2_mean": 0.15755733847618103,
      "r2_median": 0.10713684558868408,
      "r2_min": -1.071256399154663,
      "r2_max": 0.47251254320144653,
      "cosine_similarity": 0.6149669885635376
    },
    {
      "loss": 14.85929765701294,
      "mse": 14.875434875488281,
      "mae": 2.4941985607147217,
      "r2": 0.4100008010864258,
      "r2_mean": 0.16236749291419983,
      "r2_median": 0.11081355810165405,
      "r2_min": -1.0162396430969238,
      "r2_max": 0.48711633682250977,
      "cosine_similarity": 0.6155082583427429
    },
    {
      "loss": 14.847772693634033,
      "mse": 14.865653991699219,
      "mae": 2.4973011016845703,
      "r2": 0.410388708114624,
      "r2_mean": 0.16487886011600494,
      "r2_median": 0.11170411109924316,
      "r2_min": -0.7942708730697632,
      "r2_max": 0.47854679822921753,
      "cosine_similarity": 0.615551769733429
    },
    {
      "loss": 14.913341999053955,
      "mse": 14.931126594543457,
      "mae": 2.4974253177642822,
      "r2": 0.4077919125556946,
      "r2_mean": 0.15589193999767303,
      "r2_median": 0.10610753297805786,
      "r2_min": -1.1538994312286377,
      "r2_max": 0.4724867343902588,
      "cosine_similarity": 0.6148379445075989
    },
    {
      "loss": 14.88513479232788,
      "mse": 14.903257369995117,
      "mae": 2.495039701461792,
      "r2": 0.4088972806930542,
      "r2_mean": 0.15916989743709564,
      "r2_median": 0.10846221446990967,
      "r2_min": -0.6741757392883301,
      "r2_max": 0.4779927730560303,
      "cosine_similarity": 0.6153736710548401
    },
    {
      "loss": 14.853831005096435,
      "mse": 14.87480640411377,
      "mae": 2.495945453643799,
      "r2": 0.4100257158279419,
      "r2_mean": 0.16293145716190338,
      "r2_median": 0.11056971549987793,
      "r2_min": -0.6832377910614014,
      "r2_max": 0.48120754957199097,
      "cosine_similarity": 0.615666389465332
    },
    {
      "loss": 14.904703044891358,
      "mse": 14.922683715820312,
      "mae": 2.4900529384613037,
      "r2": 0.4081267714500427,
      "r2_mean": 0.1518985778093338,
      "r2_median": 0.10581398010253906,
      "r2_min": -0.7936090230941772,
      "r2_max": 0.48278993368148804,
      "cosine_similarity": 0.6155809164047241
    },
    {
      "loss": 14.888559818267822,
      "mse": 14.908379554748535,
      "mae": 2.491658926010132,
      "r2": 0.4086940884590149,
      "r2_mean": 0.15662570297718048,
      "r2_median": 0.10760921239852905,
      "r2_min": -0.7962585687637329,
      "r2_max": 0.4714916944503784,
      "cosine_similarity": 0.615370512008667
    },
    {
      "loss": 14.876879215240479,
      "mse": 14.896549224853516,
      "mae": 2.5029234886169434,
      "r2": 0.40916335582733154,
      "r2_mean": 0.1605764925479889,
      "r2_median": 0.10910224914550781,
      "r2_min": -0.7295284271240234,
      "r2_max": 0.4685606360435486,
      "cosine_similarity": 0.6157647371292114
    },
    {
      "loss": 14.884597015380859,
      "mse": 14.907272338867188,
      "mae": 2.5022642612457275,
      "r2": 0.40873801708221436,
      "r2_mean": 0.15845462679862976,
      "r2_median": 0.10882627964019775,
      "r2_min": -0.6190869808197021,
      "r2_max": 0.47576379776000977,
      "cosine_similarity": 0.6157829761505127
    },
    {
      "loss": 14.957697200775147,
      "mse": 14.977471351623535,
      "mae": 2.493288278579712,
      "r2": 0.4059537649154663,
      "r2_mean": 0.14516356587409973,
      "r2_median": 0.10180491209030151,
      "r2_min": -0.6528760194778442,
      "r2_max": 0.4667338728904724,
      "cosine_similarity": 0.6156779527664185
    },
    {
      "loss": 14.916678142547607,
      "mse": 14.93703556060791,
      "mae": 2.4903006553649902,
      "r2": 0.40755754709243774,
      "r2_mean": 0.14896678924560547,
      "r2_median": 0.10595786571502686,
      "r2_min": -0.7188646793365479,
      "r2_max": 0.4742421507835388,
      "cosine_similarity": 0.6154994368553162
    },
    {
      "loss": 14.865546321868896,
      "mse": 14.887428283691406,
      "mae": 2.5101542472839355,
      "r2": 0.4095250964164734,
      "r2_mean": 0.16052259504795074,
      "r2_median": 0.10998886823654175,
      "r2_min": -0.6197515726089478,
      "r2_max": 0.47650283575057983,
      "cosine_similarity": 0.6155655980110168
    },
    {
      "loss": 14.846877288818359,
      "mse": 14.871088981628418,
      "mae": 2.4881091117858887,
      "r2": 0.4101731777191162,
      "r2_mean": 0.1621471494436264,
      "r2_median": 0.11080384254455566,
      "r2_min": -0.5790590047836304,
      "r2_max": 0.48183560371398926,
      "cosine_similarity": 0.6159322261810303
    },
    {
      "loss": 14.829891395568847,
      "mse": 14.853211402893066,
      "mae": 2.4977495670318604,
      "r2": 0.41088223457336426,
      "r2_mean": 0.16360466182231903,
      "r2_median": 0.11254703998565674,
      "r2_min": -0.6352555751800537,
      "r2_max": 0.4856938123703003,
      "cosine_similarity": 0.6161171197891235
    },
    {
      "loss": 14.89209451675415,
      "mse": 14.911933898925781,
      "mae": 2.4939732551574707,
      "r2": 0.4085531234741211,
      "r2_mean": 0.15684914588928223,
      "r2_median": 0.1069636344909668,
      "r2_min": -0.640758752822876,
      "r2_max": 0.4741995334625244,
      "cosine_similarity": 0.6157078742980957
    },
    {
      "loss": 14.84346399307251,
      "mse": 14.868304252624512,
      "mae": 2.496739625930786,
      "r2": 0.410283625125885,
      "r2_mean": 0.1621665209531784,
      "r2_median": 0.11146819591522217,
      "r2_min": -0.6071083545684814,
      "r2_max": 0.4825822710990906,
      "cosine_similarity": 0.6160775423049927
    },
    {
      "loss": 14.857141494750977,
      "mse": 14.878520965576172,
      "mae": 2.4935288429260254,
      "r2": 0.40987837314605713,
      "r2_mean": 0.1595916897058487,
      "r2_median": 0.1102403998374939,
      "r2_min": -0.5447111129760742,
      "r2_max": 0.48092663288116455,
      "cosine_similarity": 0.616010844707489
    },
    {
      "loss": 14.838953590393066,
      "mse": 14.862383842468262,
      "mae": 2.4920639991760254,
      "r2": 0.4105184078216553,
      "r2_mean": 0.16222837567329407,
      "r2_median": 0.11205309629440308,
      "r2_min": -0.5908510684967041,
      "r2_max": 0.4897614121437073,
      "cosine_similarity": 0.6162072420120239
    },
    {
      "loss": 14.872916889190673,
      "mse": 14.895835876464844,
      "mae": 2.492220163345337,
      "r2": 0.4091916084289551,
      "r2_mean": 0.15790095925331116,
      "r2_median": 0.10900479555130005,
      "r2_min": -0.6392252445220947,
      "r2_max": 0.47985297441482544,
      "cosine_similarity": 0.6161798238754272
    },
    {
      "loss": 14.848357200622559,
      "mse": 14.869074821472168,
      "mae": 2.491971254348755,
      "r2": 0.41025304794311523,
      "r2_mean": 0.16160178184509277,
      "r2_median": 0.11103129386901855,
      "r2_min": -0.5674136877059937,
      "r2_max": 0.48249155282974243,
      "cosine_similarity": 0.6158940196037292
    },
    {
      "loss": 14.84712200164795,
      "mse": 14.870190620422363,
      "mae": 2.4879353046417236,
      "r2": 0.4102088212966919,
      "r2_mean": 0.16159400343894958,
      "r2_median": 0.11134791374206543,
      "r2_min": -0.48546934127807617,
      "r2_max": 0.4863063097000122,
      "cosine_similarity": 0.6163515448570251
    },
    {
      "loss": 14.835163497924805,
      "mse": 14.857341766357422,
      "mae": 2.4926178455352783,
      "r2": 0.4107184410095215,
      "r2_mean": 0.1633950173854828,
      "r2_median": 0.11202043294906616,
      "r2_min": -0.530930757522583,
      "r2_max": 0.48822295665740967,
      "cosine_similarity": 0.6163076162338257
    },
    {
      "loss": 14.849153804779053,
      "mse": 14.873265266418457,
      "mae": 2.489553451538086,
      "r2": 0.41008687019348145,
      "r2_mean": 0.1598329395055771,
      "r2_median": 0.11103701591491699,
      "r2_min": -0.5558837652206421,
      "r2_max": 0.48828572034835815,
      "cosine_similarity": 0.6163673996925354
    },
    {
      "loss": 14.837845039367675,
      "mse": 14.864779472351074,
      "mae": 2.492492914199829,
      "r2": 0.4104234576225281,
      "r2_mean": 0.1624802201986313,
      "r2_median": 0.11204278469085693,
      "r2_min": -0.4751960039138794,
      "r2_max": 0.48859667778015137,
      "cosine_similarity": 0.616523802280426
    },
    {
      "loss": 14.834146118164062,
      "mse": 14.86093807220459,
      "mae": 2.4950196743011475,
      "r2": 0.410575807094574,
      "r2_mean": 0.16182731091976166,
      "r2_median": 0.11182641983032227,
      "r2_min": -0.5007569789886475,
      "r2_max": 0.48115187883377075,
      "cosine_similarity": 0.6166068911552429
    },
    {
      "loss": 14.858742618560791,
      "mse": 14.88310432434082,
      "mae": 2.490678310394287,
      "r2": 0.40969663858413696,
      "r2_mean": 0.15891805291175842,
      "r2_median": 0.109546959400177,
      "r2_min": -0.5174770355224609,
      "r2_max": 0.47705698013305664,
      "cosine_similarity": 0.6164016723632812
    },
    {
      "loss": 14.834551334381104,
      "mse": 14.859620094299316,
      "mae": 2.4903762340545654,
      "r2": 0.4106280207633972,
      "r2_mean": 0.1622547209262848,
      "r2_median": 0.11145961284637451,
      "r2_min": -0.4934464693069458,
      "r2_max": 0.4819018840789795,
      "cosine_similarity": 0.6163001656532288
    },
    {
      "loss": 14.84318494796753,
      "mse": 14.866005897521973,
      "mae": 2.4943947792053223,
      "r2": 0.4103747606277466,
      "r2_mean": 0.16086776554584503,
      "r2_median": 0.11100482940673828,
      "r2_min": -0.524311900138855,
      "r2_max": 0.4847676753997803,
      "cosine_similarity": 0.6160421967506409
    },
    {
      "loss": 14.849125957489013,
      "mse": 14.874492645263672,
      "mae": 2.4943177700042725,
      "r2": 0.41003817319869995,
      "r2_mean": 0.1601875126361847,
      "r2_median": 0.11058330535888672,
      "r2_min": -0.5290346145629883,
      "r2_max": 0.4850912094116211,
      "cosine_similarity": 0.6163269877433777
    },
    {
      "loss": 14.86221399307251,
      "mse": 14.887996673583984,
      "mae": 2.492161989212036,
      "r2": 0.4095025658607483,
      "r2_mean": 0.15765288472175598,
      "r2_median": 0.10968595743179321,
      "r2_min": -0.5253614187240601,
      "r2_max": 0.48026442527770996,
      "cosine_similarity": 0.6162769198417664
    },
    {
      "loss": 14.870280551910401,
      "mse": 14.896766662597656,
      "mae": 2.4903554916381836,
      "r2": 0.4091547131538391,
      "r2_mean": 0.1571863889694214,
      "r2_median": 0.1089475154876709,
      "r2_min": -0.5120850801467896,
      "r2_max": 0.4827103018760681,
      "cosine_similarity": 0.6164094805717468
    },
    {
      "loss": 14.85051155090332,
      "mse": 14.874628067016602,
      "mae": 2.4928839206695557,
      "r2": 0.41003280878067017,
      "r2_mean": 0.15955686569213867,
      "r2_median": 0.11061054468154907,
      "r2_min": -0.5283162593841553,
      "r2_max": 0.4847540855407715,
      "cosine_similarity": 0.6161864995956421
    },
    {
      "loss": 14.84862174987793,
      "mse": 14.871295928955078,
      "mae": 2.4871699810028076,
      "r2": 0.4101649522781372,
      "r2_mean": 0.16035127639770508,
      "r2_median": 0.11071151494979858,
      "r2_min": -0.5238785743713379,
      "r2_max": 0.4887036085128784,
      "cosine_similarity": 0.616011381149292
    },
    {
      "loss": 14.8465012550354,
      "mse": 14.872186660766602,
      "mae": 2.4913101196289062,
      "r2": 0.4101296663284302,
      "r2_mean": 0.1603248119354248,
      "r2_median": 0.11069095134735107,
      "r2_min": -0.5136380195617676,
      "r2_max": 0.4795147180557251,
      "cosine_similarity": 0.6162636280059814
    }
  ]
}