| PredictionOutput(predictions=array([[-2.3132317 , 4.248802 , -1.7789735 ], | |
| [ 2.1847813 , -2.8792734 , 0.12165037], | |
| [-2.7687533 , -0.5294656 , 2.2465725 ], | |
| ..., | |
| [-1.4268643 , -2.7873724 , 3.2080243 ], | |
| [-0.053789 , 1.3824667 , -1.916969 ], | |
| [ 3.2297757 , -2.5666943 , -1.1765741 ]], dtype=float32), label_ids=array([1, 0, 2, ..., 2, 1, 0]), metrics={'test_loss': 0.4232962727546692, 'test_accuracy': 0.8401291885345176, 'test_f1': 0.8402719548154258, 'test_runtime': 19.8748, 'test_samples_per_second': 249.26, 'test_steps_per_second': 15.598}) |