Class AddScoresStageBase

Base Type

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

Derived Types

class AddScoresStageBase : public mrc::pymrc::PythonNode<std::shared_ptr<MultiResponseMessage>, std::shared_ptr<MultiResponseMessage>>

Base class for both AddScoresStage and AddClassificationStage

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

© Copyright 2023, NVIDIA. Last updated on Apr 11, 2023.