templates.MultipolarImageModelBasedTemplateSchema
- class templates.MultipolarImageModelBasedTemplateSchema(*args, **kwargs)
Model Based template
- Parameters:
model_name (str) – Model name
model_label (str) – Model label
label (str) – Label
dirpath (str) – The path to the stored model in minio.
label_map (Dict[str, int]) – The label map for the Node that this LF is being applied to
pos_threshold (float, default 0.5) – Positive threshold for model decision function. Datapoints with decision function greater than or equal are voted positive
neg_threshold (float, default -0.5) – Negative threshold for model decision function. Datapoints with decision function less than or equal are voted negative
invert_votes (bool) – Whether to invert the votes.