Skip to content

On-premise users: click in-app to access the full platform documentation for your version of DataRobot.

Select robust features by permutation importance

Access this AI accelerator on GitHub

Machine learning models have biases using small data, and some industries (e.g., healthcare and manufacturing) lack labeled data. In light of this, a good approach is to select robust features to build models. This accelerator introduces an approach to select robust features, use multiple seeds for cross validation, add dummy features to compute the median permutation importance, and select the most robust dummy features.

This notebook outlines how to:

  • Connect to DataRobot.
  • Create multiple projects by multiple seeds and add dummy features.
  • Create blender models of top-performing models.
  • Retrieve modeling permutation importance from the top-performing blender models.
  • Remove features whose permutation importance is lower than dummy features.

Updated April 2, 2024