A wonderful work! But how do you calculate the accuracy of your predictions?

Because of the encapsulation of cal_inf_loss, it is not very convenient to obtain prediction labels, it is recommended that the author add the function encapsulation of obtaining prediction labels and calculating accuracy! This will benefit users who are new to the project.