#include <ML.h>

Public Member Functions
void	init (o2::framework::ProcessingContext &pc) final
	Initialize the policy.

float	process (const TrackTRD &trk, const o2::globaltracking::RecoContainer &input, bool isTPCTRD) const final
	Calculate a PID for a given track.

Public Member Functions inherited from o2::trd::PIDBase
virtual	~PIDBase ()=default

	PIDBase (PIDPolicy policy)

void	setLocalGainFactors (const LocalGainFactor *localGain)
	Set krypton calibration.

Additional Inherited Members
Protected Member Functions inherited from o2::trd::PIDBase
std::array< float, constants::NCHARGES >	getCharges (const Tracklet64 &tracklet, const int layer, const TrackTRD &trk, const o2::globaltracking::RecoContainer &input, float snp, float tgl) const noexcept

Protected Attributes inherited from o2::trd::PIDBase
const TRDPIDParams &	mParams {TRDPIDParams::Instance()}
	parameters

const PIDPolicy	mPolicy
	policy

Detailed Description

This is the ML Base class which defines the interface all machine learning models.

Definition at line 39 of file ML.h.

Member Function Documentation

void o2::trd::ML::init ( o2::framework::ProcessingContext & pc )

finalvirtual

Initialize the policy.

Definition at line 51 of file ML.cxx.

float o2::trd::ML::process	(	const TrackTRD &	trk,
		const o2::globaltracking::RecoContainer &	input,
		bool	isTPCTRD
	)		const

finalvirtual

Calculate a PID for a given track.

Definition at line 106 of file ML.cxx.

The documentation for this class was generated from the following files: