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