MSE = 0.5542897
global_step = 538
label_ids = [5.   4.75 5.   ... 2.   0.   0.  ]
loss = 0.5557262
pearson = 0.88552225
pred = [4.8246083 4.85009   4.833224  ... 1.9331673 0.7744228 1.5018901]
spearman = 0.77637976
