| 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}) |