Method target

diffpack/task.py:142–146

(self, batch)

        return pred_score, score_norm

    def target(self, batch):
        protein = batch["graph"]
        target_score = batch['score']  # Move to protein attribute
        target_score = target_score * protein.chi_mask
        return target_score

    def evaluate(self, pred, target):
        metric = {}