estimator package

Generic Estimator and FrameworkBaseEstimator-related files.

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.

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.

MMLBaseEstimator

Abstract base class for all estimators.

MMLBaseEstimatorRunConfig

Abstract base class for all Estimator run configs.

DEPRECATED. Use the RunConfiguration class.

Mpi

Manages MPI settings for distributed training jobs.

ParameterServer

Manages Parameter Server settings for training jobs.

Gloo

A class for managing Gloo settings for jobs.

Nccl

A class for managing Gloo settings for jobs.