PredictionOutput(predictions=array([[ 0.9608663 ,  0.22863041, -2.0746882 ],
       [ 1.5385464 , -1.1549778 , -1.4723089 ],
       [-0.5163633 ,  0.5942186 , -0.63961476],
       ...,
       [-2.0814145 ,  4.172013  , -1.7399795 ],
       [ 2.2152631 , -0.09021759, -2.9525821 ],
       [-1.9521435 ,  3.2808514 , -1.162269  ]], dtype=float32), label_ids=array([1, 0, 0, ..., 1, 0, 1]), metrics={'test_loss': 0.597151517868042, 'test_accuracy': 0.7891996142719383, 'test_f1': 0.7908201752093549, 'test_runtime': 62.9177, 'test_samples_per_second': 82.409, 'test_steps_per_second': 5.165})