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(**kwargs)

Base Class for a Machine Learning Based Reconstructor.

SKLearnRegressionReconstructor(**kwargs)

Base class for regression tasks.

SKLearnClassificationReconstructor(**kwargs)

Base class for classification tasks.

EnergyRegressor(**kwargs)

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

ParticleClassifier(**kwargs)

Predict dl2 particle classification.

DispReconstructor(**kwargs)

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

CrossValidator(**kwargs)

Class to train sklearn based reconstructors in a cross validation.