Class AddScoresStage

Base Type

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

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

Add probability scores to each message. Score 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

AddScoresStage(std::size_t num_class_labels, std::map<std::size_t, std::string> idx2label)

Construct a new Add Scores Stage object.

Parameters
  • num_class_labels – : Number of classification labels

  • idx2label – : Index to classification labels map

subscribe_fn_t build_operator()

TODO(Documentation)

Public Members

std::size_t m_num_class_labels
std::map<std::size_t, std::string> m_idx2label

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.