| PredictionOutput(predictions=array([[ 0.16208035, 1.5503061 , -1.7306447 ], | |
| [ 0.35421252, -0.64115864, 0.3081979 ], | |
| [-1.7991027 , 1.6296394 , 0.32940128], | |
| ..., | |
| [-1.7003723 , 2.8618367 , -0.79798317], | |
| [ 2.0220978 , -0.26142257, -1.6351734 ], | |
| [-1.511989 , 2.713712 , -0.7102815 ]], dtype=float32), label_ids=array([1, 0, 0, ..., 1, 0, 1]), metrics={'test_loss': 0.7916349172592163, 'test_accuracy': 0.7363548698167792, 'test_f1': 0.736074217140886, 'test_runtime': 62.9352, 'test_samples_per_second': 82.386, 'test_steps_per_second': 5.164}) |