Skip to content

ignore-cols for mlnet not working as it includes all other input columns to features #2485

Description

@cyberkoolman

System Information (please complete the following information):

  • OS & Version: Windows 10 & Windows 11
  • ML.NET Version: ML.NET v2.0.1
  • .NET Version: .NET 7.0

Describe the bug
Using the mlnet cli on wikiDetoxAnnotated40kRows.tsv which has the columns of Label, rev_id, comment, year, logged_in, ns, sample, split, wanted to pick only comment as the input feature. Currently, through CLI, have to use --ignore-cols parameter and resulted the training pipeline to include all the columns to feature. I had to resort to VS ModelBuilder for go around this issue for now.

To Reproduce
Execute cli as below
mlnet classification --dataset .\wikiDetoxAnnotated40kRows.tsv --label-col Label --has-header true --ignore-cols rev_id year logged_in ns sample split --name Sentiment

Expected behavior
I would expect the features only contain "comment" field

Screenshots, Code, Sample Projects
Resulting training pipeline as below
image

Additional context

  1. Shouldn't this be easier if providing the option to choose only relevant column, instead of long list of ignore-columns.
  2. Documentation says ignore-columns, not matching to ignore-cols. [https://learn.microsoft.com/en-us/dotnet/machine-learning/reference/ml-net-cli-reference#ignore-columns]

Metadata

Metadata

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions