tumourkit.classification.train_graphs.evaluate

tumourkit.classification.train_graphs.evaluate(loader: GraphDataLoader, model: Module, device: str, writer: SummaryWriter | None = None, epoch: int | None = None, log_suffix: str | None = None, num_classes: str | None = 2, enable_background: bool | None = False) → List[float]

Evaluates model in loader. Logs to tensorboard with suffix log_suffix. Returns the model in evaluation mode.