Class AddClassificationsStage

• Defined in File add_classification.hpp

Base Type

• public mrc::pymrc::PythonNode< std::shared_ptr< MultiResponseProbsMessage >, std::shared_ptr< MultiResponseProbsMessage > >

class AddClassificationsStage : public mrc::pymrc::PythonNode<std::shared_ptr<MultiResponseProbsMessage>, std::shared_ptr<MultiResponseProbsMessage>>

Add detected classifications to each message. Classification labels based on probabilities calculated in inference stage. Label indexes will be looked up in the idx2label property.

Public Types

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

Public Functions

AddClassificationsStage(float threshold, std::size_t num_class_labels, std::map<std::size_t, std::string> idx2label)

Construct a new Add Classifications Stage object.

Parameters

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

• num_class_labels – : Number of classification labels

• idx2label – : Index to classification labels map