estimator package

Contains base estimator classes and the generic estimator class in Azure Machine Learning.

Classes

Estimator

Represents a generic estimator to train data using any supplied framework.

This class is designed for use with machine learning frameworks that do not already have an Azure Machine Learning pre-configured estimator. Pre-configured estimators exist for Chainer, PyTorch, TensorFlow, and SKLearn. To create an Estimator that is not preconfigured, see Train models with Azure Machine Learning using estimator.

The Estimator class wraps run configuration information to help simplify the tasks of specifying how a script is executed. It supports single-node as well as multi-node execution. Running the estimator produces a model in the output directory specified in your training script.

Gloo

Manages Gloo settings for distributed training jobs.

Gloo can be specified for a training job with the distributed_training parameter of the preconfigured PyTorch estimator or any generic Estimator supporting Gloo.

MMLBaseEstimator

Abstract base class for all estimators.

MMLBaseEstimatorRunConfig

Abstract base class for all Estimator run configs.

DEPRECATED. Use the RunConfiguration class.

Mpi

Manages Message Passing Interface (MPI) settings for distributed training jobs.

MPI can be specified for a job with the distributed_training parameter of preconfigured estimators Chainer, PyTorch, and TensorFlow, or with a generic Estimator.

Nccl

Manages Nccl settings for distributed training jobs.

Nccl can be specified for a training job with the distributed_training parameter of the preconfigured PyTorch estimator or any generic Estimator supporting Nccl.

ParameterServer

Manages Parameter Server settings for training jobs.

DEPRECATED. Use the TensorflowConfiguration class.