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.