Class AddScoresStage#

Inheritance Relationships#

Base Type#

Class Documentation#

class AddScoresStage : public morpheus::AddScoresStageBase#

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 Functions

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

Construct a new Add Scores Stage object.

Parameters:

idx2label – : Index to classification labels map