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