Struct AddClassificationStageInterfaceProxy
Defined in File add_classification.hpp
-
struct AddClassificationStageInterfaceProxy
Interface proxy, used to insulate python bindings.
Public Static Functions
-
static std::shared_ptr<mrc::segment::Object<AddClassificationsStage<MultiResponseMessage, MultiResponseMessage>>> init_multi(mrc::segment::Builder &builder, const std::string &name, std::map<std::size_t, std::string> idx2label, float threshold)
Create and initialize a AddClassificationStage that receives MultiResponseMessage and emits MultiResponseMessage, and return the result.
- Parameters
builder – : Pipeline context object reference
name – : Name of a stage reference
idx2label – : Index to classification labels map
threshold – : Threshold to consider true/false for each class
- Returns
std::shared_ptr<mrc::segment::Object<AddClassificationsStage<MultiResponseMessage, MultiResponseMessage>>>
-
static std::shared_ptr<mrc::segment::Object<AddClassificationsStage<ControlMessage, ControlMessage>>> init_cm(mrc::segment::Builder &builder, const std::string &name, std::map<std::size_t, std::string> idx2label, float threshold)
Create and initialize a AddClassificationStage that receives ControlMessage and emits ControlMessage, and return the result.
- Parameters
builder – : Pipeline context object reference
name – : Name of a stage reference
idx2label – : Index to classification labels map
threshold – : Threshold to consider true/false for each class
- Returns
std::shared_ptr<mrc::segment::Object<AddClassificationsStage<ControlMessage, ControlMessage>>>
-
static std::shared_ptr<mrc::segment::Object<AddClassificationsStage<MultiResponseMessage, MultiResponseMessage>>> init_multi(mrc::segment::Builder &builder, const std::string &name, std::map<std::size_t, std::string> idx2label, float threshold)