Skip to content

PFI (Permutation Feature Importance) takes forever on some regression models created using AutoML #7345

@lolsoftware

Description

@lolsoftware

System Information

  • Windows 11
  • ML.NET v4.0.0
  • .NET 8.0

Describe the bug
Calculating PFI seems to hang (take forever) for some regression models created using AutoML. For other models (created against the same learning data) calculating PFI takes just a couple of seconds.

To Reproduce
Run the attached program.

Expected behavior
I would expect PFI to complete in a timely fashion. Or at least provide a mechanism to monitor the progress of PFI calculation with the ability to cancel it.

Screenshots, Code, Sample Projects
The attached program contains two regression models created using AutoML, and a set of data used to create the two models. The program first calculates PFI against the first model which takes just a couple of seconds. Then the program tries to calculate PFI against the second model which takes forever.
test project.zip

Metadata

Metadata

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions