netmap.model.train_model

Autoencoder ensemble training utilities.

This module provides functions to train a zoo of independent autoencoder models with early stopping, forming the ensemble used by the GRN inference stage. Training is hardcoded to CUDA GPU.

Functions

create_model_zoo

Create an ensemble of autoencoders (model zoo) trained on the supplied data.