modified the accuracy layer so that it can export the predicted labels