otx.algorithms.classification.task#

Task of OTX Classification.

Classes

OTXClassificationTask(task_environment[, ...])

Task class for OTX classification.
class otx.algorithms.classification.task.OTXClassificationTask(task_environment: TaskEnvironment, output_path: str | None = None)[source]#

Bases: OTXTask, ABC

Task class for OTX classification.

evaluate(output_resultset: ResultSetEntity, evaluation_metric: str | None = None)[source]#

Evaluate function of OTX Classification Task.

explain(dataset: DatasetEntity, explain_parameters: ExplainParameters | None = None) → DatasetEntity[source]#

Main explain function of OTX Classification Task.

export(export_type: ExportType, output_model: ModelEntity, precision: ModelPrecision = ModelPrecision.FP32, dump_features: bool = True)[source]#

Export function of OTX Classification Task.

infer(dataset: DatasetEntity, inference_parameters: InferenceParameters | None = None) → DatasetEntity[source]#

Main infer function of OTX Classification.

save_model(output_model: ModelEntity)[source]#

Save best model weights in ClassificationTrainTask.

train(dataset: DatasetEntity, output_model: ModelEntity, train_parameters: TrainParameters | None = None, seed: int | None = None, deterministic: bool = False)[source]#

Train function for OTX classification task.

Actual training is processed by _train_model fucntion