def loss_audit(self, x, s, f, w): s_pred = self.forward_s(x, f) loss = weighted_cross_entropy(w, s, s_pred) return loss
def loss_prediction(self, x, y, w): y_pred = self.forward_y(x) loss = weighted_cross_entropy(w, y, y_pred) return loss