Class AddScoresStageBase
Defined in File add_scores_stage_base.hpp
Base Type
public mrc::pymrc::PythonNode< std::shared_ptr< MultiResponseMessage >, std::shared_ptr< MultiResponseMessage > >
Derived Types
public morpheus::AddClassificationsStage
(Class AddClassificationsStage)public morpheus::AddScoresStage
(Class AddScoresStage)
-
class AddScoresStageBase : public mrc::pymrc::PythonNode<std::shared_ptr<MultiResponseMessage>, std::shared_ptr<MultiResponseMessage>>
Base class for both
AddScoresStage
andAddClassificationStage
Subclassed by morpheus::AddClassificationsStage, morpheus::AddScoresStage
Public Types
- using base_t = mrc::pymrc::PythonNode<std::shared_ptr<MultiResponseMessage>, std::shared_ptr<MultiResponseMessage>>
Public Functions
-
AddScoresStageBase(std::map<std::size_t, std::string> idx2label, std::optional<float> threshold)
Construct a new Add Classifications Stage object.
- Parameters
threshold – : Threshold to consider true/false for each class
idx2label – : Index to classification labels map