Share via


BinaryClassificationCatalog Class

Definition

Class used by MLContext to create instances of binary classification components, such as trainers and calibrators.

C#
public sealed class BinaryClassificationCatalog : Microsoft.ML.TrainCatalogBase
Inheritance
BinaryClassificationCatalog

Properties

Calibrators

The list of calibrators for performing binary classification.

Trainers

The list of trainers for performing binary classification.

Methods

ChangeModelThreshold<TModel>(BinaryPredictionTransformer<TModel>, Single)

Method to modify the threshold to existing model and return modified model.

CrossValidate(IDataView, IEstimator<ITransformer>, Int32, String, String, Nullable<Int32>)

Run cross-validation over numberOfFolds folds of data, by fitting estimator, and respecting samplingKeyColumnName if provided. Then evaluate each sub-model against labelColumnName and return a CalibratedBinaryClassificationMetrics object, which includes probability-based metrics, for each sub-model. Each sub-model is evaluated on the cross-validation fold that it did not see during training.

CrossValidateNonCalibrated(IDataView, IEstimator<ITransformer>, Int32, String, String, Nullable<Int32>)

Run cross-validation over numberOfFolds folds of data, by fitting estimator, and respecting samplingKeyColumnName if provided. Then evaluate each sub-model against labelColumnName and return a BinaryClassificationMetrics object, which do not include probability-based metrics, for each sub-model. Each sub-model is evaluated on the cross-validation fold that it did not see during training.

Evaluate(IDataView, String, String, String, String)

Evaluates scored binary classification data.

EvaluateNonCalibrated(IDataView, String, String, String)

Evaluates scored binary classification data, without probability-based metrics.

Extension Methods

Applies to

Product Versions
ML.NET 1.0.0, 1.1.0, 1.2.0, 1.3.1, 1.4.0, 1.5.0, 1.6.0, 1.7.0, 2.0.0, 3.0.0, Preview, 4.0.0