PredictionOutput(predictions=array([[-2.1687298 ,  3.545325  , -2.093927  ],
       [ 2.4960866 , -1.8406624 ,  0.08127579],
       [-2.714672  ,  1.8125485 , -0.01572387],
       ...,
       [-0.14208438, -0.6421326 ,  1.7041954 ],
       [-0.06215704,  1.692547  , -1.4649454 ],
       [ 2.861244  , -1.13619   , -1.0870064 ]], dtype=float32), label_ids=array([1, 0, 2, ..., 2, 1, 0]), metrics={'test_loss': 0.4816402196884155, 'test_accuracy': 0.8104561970125151, 'test_f1': 0.8108616645364378, 'test_runtime': 14.9045, 'test_samples_per_second': 332.382, 'test_steps_per_second': 20.799})