Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

allow passing main arguments to lightgbm::lgb.train #8

Closed
simonpcouch opened this issue May 2, 2022 · 1 comment · Fixed by #16
Closed

allow passing main arguments to lightgbm::lgb.train #8

simonpcouch opened this issue May 2, 2022 · 1 comment · Fixed by #16

Comments

@simonpcouch
Copy link
Contributor

simonpcouch commented May 2, 2022

At the moment, train_lightgbm only passes elements from set_engine to elements in the params argument to lightgbm::lgb.train. This seems to be where most parameter adjustments happen, but some of the main arguments would be nice to be able to set as well.

library(bonsai)
data(penguins, package = "modeldata")

m <- boost_tree() %>%
  set_engine("lightgbm", nrounds = 100) %>%
  set_mode("regression") %>%
  fit(bill_length_mm ~ ., data = penguins)
#> [LightGBM] [Warning] Unknown parameter: nrounds

Created on 2022-05-02 by the reprex package (v2.0.1)

@github-actions
Copy link

This issue has been automatically locked. If you believe you have found a related problem, please file a new issue (with a reprex: https://reprex.tidyverse.org) and link to this issue.

@github-actions github-actions bot locked and limited conversation to collaborators Jan 11, 2023
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant