Skip to content

AutoML Featurizer handling of label column #6400

Open

Description

@luisquintanilla

Problem

Currently when using the Featurizer with ColumnInformation from ColumnInferenceResults there is no easy way to exclude the label column from featurization or automatically featurize it as well. Therefore you either need to pass in an IDataView with the dropped columns or you need to individually provide the column type parameters (i.e. numericColumns).

var pipeline =
    ctx.Auto().Featurizer(
        data: dataView,
        numericColumns: columnInferenceResults.ColumnInformation.NumericColumnNames.ToArray(),
        outputColumnName: "Features",
        excludeColumns: new[] { "Label" })
        .Append(ctx.Auto().MultiClassification());

Proposal

  1. Automatically featurize label column (this is often the case for classification scenarios where values need to be converted to KeyType).

  2. Create an overload of Featurizer with the following signature:

Featurizer(IDataView data, ColumnInformation columnInformation, string[] excludeColumns, string outputColumnName)

Activity

  1. ghost added
    untriagedNew issue has not been triaged
    on Oct 21, 2022
  2. added this to the ML.NET 3.0 milestone on Nov 28, 2022
  3. ghost removed
    untriagedNew issue has not been triaged
    on Nov 28, 2022
  4. michaelgsharp commented on Nov 28, 2022

    @michaelgsharp
    Contributor

    Moving to 3.0 milestone, we can push it out further if needed.

  5. added
    area-AutoMLAutomated machine learning, Model Builder, and ML.NET CLI
    on Sep 8, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    area-AutoMLAutomated machine learning, Model Builder, and ML.NET CLIenhancementNew feature or requestyellowmaintenance only

    Type

    No type

    Projects

    No projects

      Milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions