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 for each telescope.

CrossValidator(**kwargs)

Class to train sklearn based reconstructors in a cross validation