| PredictionOutput(predictions=array([[ 1.8757781 , 0.10563973, -1.8769046 ], | |
| [ 2.5787184 , -1.5079138 , -1.0805748 ], | |
| [ 0.2447584 , 1.4224131 , -1.8482531 ], | |
| ..., | |
| [-1.7147496 , -3.451846 , 4.9482317 ], | |
| [-3.355991 , -2.0938346 , 4.969881 ], | |
| [-2.535666 , -2.9212089 , 5.2116556 ]], dtype=float32), label_ids=array([1, 0, 0, ..., 2, 2, 2]), metrics={'test_loss': 0.5056426525115967, 'test_accuracy': 0.8241831149444352, 'test_f1': 0.8267838848593851, 'test_runtime': 89.8547, 'test_samples_per_second': 67.097, 'test_steps_per_second': 4.196}) |