Template Class AddScoresStageBase

• Defined in File add_scores_stage_base.hpp

Base Type

• public mrc::pymrc::PythonNode< std::shared_ptr< InputT >, std::shared_ptr< OutputT > >

Derived Types

• public morpheus::AddClassificationsStage< InputT, OutputT > (Template Class AddClassificationsStage)

• public morpheus::AddScoresStage< InputT, OutputT > (Template Class AddScoresStage)

template<typename InputT, typename OutputT>
class AddScoresStageBase : public mrc::pymrc::PythonNode<std::shared_ptr<InputT>, std::shared_ptr<OutputT>>

Base class for both AddScoresStage and AddClassificationStage

Subclassed by morpheus::AddClassificationsStage< InputT, OutputT >, morpheus::AddScoresStage< InputT, OutputT >

Public Types

using base_t = mrc::pymrc::PythonNode<std::shared_ptr<InputT>, std::shared_ptr<OutputT>>

Public Functions

AddScoresStageBase(std::map<std::size_t, std::string> idx2label, std::optional<float> threshold)

Construct a new Add Classifications Stage object.

Parameters

• threshold – : Threshold to consider true/false for each class

• idx2label – : Index to classification labels map

source_type_t on_data(sink_type_t x)

Called every time a message is passed to this stage