deep_tau::DeepTauBase::Output Struct Reference

#include <DeepTauBase.h>

Public Types
using	ResultMap = std::map< std::string, std::unique_ptr< TauDiscriminator >>

Public Member Functions
ResultMap	get_value (const edm::Handle< TauCollection > &taus, const tensorflow::Tensor &pred, const WPMap &working_points) const
	Output (const std::vector< size_t > &num, const std::vector< size_t > &den)

Public Attributes
std::vector< size_t >	den_
std::vector< size_t >	num_

Detailed Description

Definition at line 72 of file DeepTauBase.h.

Member Typedef Documentation

using deep_tau::DeepTauBase::Output::ResultMap = std::map<std::string, std::unique_ptr<TauDiscriminator>>

Definition at line 73 of file DeepTauBase.h.

Constructor & Destructor Documentation

deep_tau::DeepTauBase::Output::Output ( const std::vector< size_t > &  num, const std::vector< size_t > &  den  )

inline

Definition at line 76 of file DeepTauBase.h.

References das::get_value(), nano_cff::taus, and FWLite::working_points.

: num_(num), den_(den) {}

Member Function Documentation

DeepTauBase::Output::ResultMap deep_tau::DeepTauBase::Output::get_value	(	const edm::Handle< TauCollection > &	taus,
		const tensorflow::Tensor &	pred,
		const WPMap &	working_points
	)		const

Definition at line 50 of file DeepTauBase.cc.

References SiStripPI::max, convertSQLitetoXML_cfg::output, and metsig::tau.

{
    ResultMap output;
    output[""] = std::make_unique<TauDiscriminator>(TauRefProd(taus));
    for(const auto& wp : working_points)
        output[wp.first] = std::make_unique<TauDiscriminator>(TauRefProd(taus));

    for(size_t tau_index = 0; tau_index < taus->size(); ++tau_index) {
        float x = 0;
        for(size_t num_elem : num_)
            x += pred.matrix<float>()(tau_index, num_elem);
        if(x != 0 && !den_.empty()) {
            float den_val = 0;
            for(size_t den_elem : den_)
                den_val += pred.matrix<float>()(tau_index, den_elem);
            x = den_val != 0 ? x / den_val : std::numeric_limits<float>::max();
        }
        output[""]->setValue(tau_index, x);
        for(const auto& wp : working_points) {
            const auto& tau = taus->at(tau_index);
            const bool pass = x > (*wp.second)(tau);
            output[wp.first]->setValue(tau_index, pass);
        }
    }
    return output;
}

Member Data Documentation

std::vector<size_t> deep_tau::DeepTauBase::Output::den_

Definition at line 74 of file DeepTauBase.h.

std::vector<size_t> deep_tau::DeepTauBase::Output::num_

Definition at line 74 of file DeepTauBase.h.