PredictionOutput(predictions=array([[ 1.8757781 ,  0.10563973, -1.8769046 ],
       [ 2.5787184 , -1.5079138 , -1.0805748 ],
       [ 0.2447584 ,  1.4224131 , -1.8482531 ],
       ...,
       [-1.7147496 , -3.451846  ,  4.9482317 ],
       [-3.355991  , -2.0938346 ,  4.969881  ],
       [-2.535666  , -2.9212089 ,  5.2116556 ]], dtype=float32), label_ids=array([1, 0, 0, ..., 2, 2, 2]), metrics={'test_loss': 0.5056426525115967, 'test_accuracy': 0.8241831149444352, 'test_f1': 0.8267838848593851, 'test_runtime': 89.8547, 'test_samples_per_second': 67.097, 'test_steps_per_second': 4.196})