change loss function of the classifier