[78ef36]: / docs-source / source / model.rst

Download this file

53 lines (44 with data), 1.8 kB

slideflow.model

This module provides the :class:`ModelParams` class to organize model and training parameters/hyperparameters and assist with model building, as well as the :class:`Trainer` class that executes model training and evaluation. :class:`RegressionTrainer` and :class:`SurvivalTrainer` are extensions of this class, supporting regression and Cox Proportional Hazards outcomes, respectively. The function :func:`build_trainer` can choose and return the correct model instance based on the provided hyperparameters.

Note

In order to support both Tensorflow and PyTorch backends, the :mod:`slideflow.model` module will import either :mod:`slideflow.model.tensorflow` or :mod:`slideflow.model.torch` according to the currently active backend, indicated by the environmental variable SF_BACKEND.

See :ref:`training` for a detailed look at how to train models.

Trainer

RegressionTrainer

SurvivalTrainer

Features

Other functions