Module epiclass.core

The core module contains all crucial code files to the project.

It contains no directly executable script (python main.py), only code modules.

Sub-modules

epiclass.core.analysis

Module containing result analysis code.

epiclass.core.confusion_matrix

ConfusionMatrixWriter class

epiclass.core.data

Module to define data/datasets processing and representation classes.

epiclass.core.data_source

Module for reading source data files.

epiclass.core.epiatlas_treatment

Functions to split epiatlas datasets properly, keeping track types together in the different sets.

epiclass.core.estimators

Module for wrappers around simple sklearn machine learning estimators.

epiclass.core.hdf5_loader

Module for hdf5 loading handling.

epiclass.core.lgbm

Module to define how LightGBM is handled.

epiclass.core.metadata

Module from Metadata class and HealthyCategory.

epiclass.core.model_pytorch

Model creation module

epiclass.core.shap_values

Module containing shap values related code (e.g. handling computation, analysing results).

epiclass.core.spectro

epiclass.core.trainer

Trainer class extensions module

epiclass.core.types

Define types useful for the project.