Implements the meta classes for training and inference with deep model ensembles for deepflash2.

Base class

class EnsembleBase[source]

EnsembleBase(image_dir:str=None, mask_dir:str=None, files:List[Path]=None, label_fn:callable=None, config:Config=None, path:Path=None, zarr_store:str=None) :: GetAttr

Inherit from this to have all attr accesses in self._xtra passed down to self.default

tst = EnsembleBase()

Ensemble Learner

class EnsembleLearner[source]

EnsembleLearner(*args, ensemble_path=None, preproc_dir=None, metrics=None, cbs=None, ds_kwargs={}, dl_kwargs={}, model_kwargs={}, stats=None, **kwargs) :: EnsembleBase

Meta class to training model ensembles with n models

Ensemble Prediction Class

class EnsemblePredictor[source]

EnsemblePredictor(*args, ensemble_path:Path=None, **kwargs) :: EnsembleBase

Inherit from this to have all attr accesses in self._xtra passed down to self.default

t = EnsemblePredictor()