Reconstructors Based On Scikit-Learn (sklearn)#

Introduction#

This module provides Reconstructor implementations based on scikit-learn machine learning algorithms.

Reference/API#

ctapipe.reco.sklearn Module#

Component Wrappers around sklearn models

Classes#

SKLearnReconstructor(*args, **kwargs)

Base Class for a Machine Learning Based Reconstructor.

SKLearnRegressionReconstructor(*args, **kwargs)

Base class for regression tasks.

SKLearnClassificationReconstructor(*args, ...)

Base class for classification tasks.

EnergyRegressor(*args, **kwargs)

Use a scikit-learn regression model per telescope type to predict primary energy.

ParticleClassifier(*args, **kwargs)

Predict dl2 particle classification.

DispReconstructor(*args, **kwargs)

Predict absolute value and sign for disp origin reconstruction and convert to altitude and azimuth prediction for each telescope.

CrossValidator(*args, **kwargs)

Class to train sklearn based reconstructors in a cross validation.