PredictionOutput(predictions=array([[ 0.16208035,  1.5503061 , -1.7306447 ],
       [ 0.35421252, -0.64115864,  0.3081979 ],
       [-1.7991027 ,  1.6296394 ,  0.32940128],
       ...,
       [-1.7003723 ,  2.8618367 , -0.79798317],
       [ 2.0220978 , -0.26142257, -1.6351734 ],
       [-1.511989  ,  2.713712  , -0.7102815 ]], dtype=float32), label_ids=array([1, 0, 0, ..., 1, 0, 1]), metrics={'test_loss': 0.7916349172592163, 'test_accuracy': 0.7363548698167792, 'test_f1': 0.736074217140886, 'test_runtime': 62.9352, 'test_samples_per_second': 82.386, 'test_steps_per_second': 5.164})