MolecularDiffusion.modules.tasks.task

Classes

Task

Module Contents

class MolecularDiffusion.modules.tasks.task.Task

Bases: torch.nn.Module

abstractmethod evaluate(pred, target)

abstractmethod predict(batch, all_loss=None, metric=None)

predict_and_target(batch, all_loss=None, metric=None)

preprocess(train_set, valid_set=None, test_set=None)

abstractmethod target(batch)