Compocyte.core.models.dense_torchΒΆ

PyTorch dense model wrapper for Compocyte.

This module wraps PyTorch-based feed-forward models used as local classifiers inside the hierarchical framework. It provides loading/saving helpers and a standard interface for training and prediction used by the higher-level code.

class Compocyte.core.models.dense_torch.DenseTorch(*args: Any, **kwargs: Any)[source]ΒΆ

Bases: Module

forward(x)[source]ΒΆ
classmethod import_external(model, labels)[source]ΒΆ
predict(x) numpy.array[source]ΒΆ
predict_logits(x) numpy.array[source]ΒΆ
reset_output(n_output)[source]ΒΆ