File size: 592 Bytes
84be7df |
1 2 3 4 5 6 7 |
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}) |