Template Class AddScoresStage

Base Type

template<typename InputT, typename OutputT>
class AddScoresStage : public morpheus::AddScoresStageBase<InputT, OutputT>

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.


idx2label – : Index to classification labels map

Previous Template Class AddClassificationsStage
Next Template Class AddScoresStageBase
© Copyright 2024, NVIDIA. Last updated on Apr 25, 2024.