tidymodels / tidymodels/dials

Tuning of imputation hyperparameters

Open
#154 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

feature
Dominant language
R
Stars
117
Forks
36
Avg merge
1h 53m
Merged PRs (30d)
2

Description

Feature

In situations when I am imputing data as a recipe step, I would like to be able to tune the hyperparameters of the imputation method - which I anticipate is going to be a nightmare to implement, especially when parameters are passed between packages. For example, imputing with step_bagimpute uses ipred, which has its own hyperparameters but also passes on hyperparameters to rpart!

# the ipred hyperparameters, e.g. nbagg, can be set:

step_bagimpute(options = list(nbagg = 25))

# I'd like to be able to do something like:

step_bagimpute(options = list(nbagg = tune()))

But there's another level of HPs that ipred passes on to rpart, and it would be nice to be able to tune them as well.

# HPs can be passed through ipred to rpart

step_bagimpute(options = list(control = rpart::rpart.control(minsplit = 10))

# but it would be great to tune even these "next level" parameters:

step_bagimpute(options = list(control = rpart::rpart.control(minsplit = tune()))

Good luck!

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the step_bagimpute examples in the issue and inspect how dials represents tuneable parameters across the ipred and rpart::rpart.control entry points. Done would mean supporting tune() for ipred options such as nbagg and nested rpart parameters such as minsplit, with the resulting values available for tuning.

Written by the indexing model from the issue text.

Assessment

Tech stack
r
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.