zoo.orca.learn package¶
Subpackages¶
- zoo.orca.learn.horovod package
- zoo.orca.learn.mxnet package
- zoo.orca.learn.pytorch package
- Submodules
- zoo.orca.learn.pytorch.constants module
- zoo.orca.learn.pytorch.estimator module
- zoo.orca.learn.pytorch.pytorch_horovod_estimator module
- zoo.orca.learn.pytorch.pytorch_trainer module
- zoo.orca.learn.pytorch.torch_runner module
- zoo.orca.learn.pytorch.training_operator module
- zoo.orca.learn.pytorch.utils module
- Module contents
- zoo.orca.learn.tf package
- zoo.orca.learn.tf2 package
Submodules¶
zoo.orca.learn.metrics module¶
-
class
zoo.orca.learn.metrics.AUC(threshold_num=200)[source]¶ Bases:
zoo.orca.learn.metrics.MetricsMetric for binary(0/1) classification, support single label and multiple labels.
# Arguments threshold_num: The number of thresholds. The quality of approximation
may vary depending on threshold_num.>>> meter = AUC(20)
-
class
zoo.orca.learn.metrics.Accuracy(zero_based_label=True)[source]¶ Bases:
zoo.orca.learn.metrics.MetricsMeasures top1 accuracy for multi-class classification or accuracy for binary classification.
# Arguments zero_based_label: Boolean. Whether target labels start from 0. Default is True.
If False, labels start from 1. Note that this only takes effect for multi-class classification. For binary classification, labels ought to be 0 or 1.>>> acc = Accuracy()
-
class
zoo.orca.learn.metrics.BinaryAccuracy[source]¶ Bases:
zoo.orca.learn.metrics.MetricsMeasures top1 accuracy for binary classification with zero-based index.
>>> acc = BinaryAccuracy()
-
class
zoo.orca.learn.metrics.CategoricalAccuracy[source]¶ Bases:
zoo.orca.learn.metrics.MetricsMeasures top1 accuracy for multi-class classification when target is one-hot encoded.
>>> acc = CategoricalAccuracy()
-
class
zoo.orca.learn.metrics.MAE[source]¶ Bases:
zoo.orca.learn.metrics.MetricsMetric for mean absoluate error, similar from MAE criterion
>>> mae = MAE()
-
class
zoo.orca.learn.metrics.SparseCategoricalAccuracy[source]¶ Bases:
zoo.orca.learn.metrics.MetricsMeasures top1 accuracy for multi-class classification with sparse target.
>>> acc = SparseCategoricalAccuracy()
-
class
zoo.orca.learn.metrics.Top5Accuracy[source]¶ Bases:
zoo.orca.learn.metrics.MetricsMeasures top5 accuracy for multi-class classification.
# Arguments zero_based_label: Boolean. Whether target labels start from 0. Default is True.
If False, labels start from 1.>>> acc = Top5Accuracy()
zoo.orca.learn.trigger module¶
-
class
zoo.orca.learn.trigger.EveryEpoch[source]¶ Bases:
zoo.orca.learn.trigger.TriggerA trigger specifies a timespot or several timespots during training, and a corresponding action will be taken when the timespot(s) is reached. EveryEpoch is a trigger that triggers an action when each epoch finishs. Could be used as trigger in setvalidation and setcheckpoint in Optimizer, and also in TrainSummary.set_summary_trigger.
>>> everyEpoch = EveryEpoch()
-
class
zoo.orca.learn.trigger.SeveralIteration(interval)[source]¶ Bases:
zoo.orca.learn.trigger.TriggerA trigger specifies a timespot or several timespots during training, and a corresponding action will be taken when the timespot(s) is reached. SeveralIteration is a trigger that triggers an action every “n” iterations. Could be used as trigger in setvalidation and setcheckpoint in Optimizer, and also in TrainSummary.set_summary_trigger.
>>> serveralIteration = SeveralIteration(2)