Template Class AddScoresStage
Defined in File add_scores.hpp
Base Type
public morpheus::AddScoresStageBase< InputT, OutputT >
(Template Class AddScoresStageBase)
-
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.
- Parameters
idx2label – : Index to classification labels map
-
AddScoresStage(std::map<std::size_t, std::string> idx2label)