np.round() function rounds off to nearest value what if I get different values in the output tensor like tensor([-3.44,-2.678,-0.65,0.96])
then after rounding I get array([-3,-2,-0,1]) but for accuracy_score the values should be in 0 and 1