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

setting stop_iter with brulee model #1050

Closed
topepo opened this issue Jan 16, 2024 · 1 comment · Fixed by #1051
Closed

setting stop_iter with brulee model #1050

topepo opened this issue Jan 16, 2024 · 1 comment · Fixed by #1051
Assignees

Comments

@topepo
Copy link
Member

topepo commented Jan 16, 2024

library(tidymodels)
# requires brulee; make sure to load it after install to get extra downloads
set.seed(1)
mlp(hidden_units = 5, epochs = 100) %>% 
  set_engine("brulee", stop_iter = 5)  %>% 
  set_mode("regression") %>% 
  fit(price ~ latitude + longitude, data = Sacramento)
#> Warning: The following arguments cannot be manually modified and were removed:
#> stop_iter.
#> parsnip model object
#> 
#> Multilayer perceptron
#> 
#> relu activation
#> 5 hidden units,  21 model parameters
#> 932 samples, 2 features, numeric outcome 
#> weight decay: 0.001 
#> dropout proportion: 0 
#> batch size: 839 
#> learn rate: 0.01 
#> scaled validation loss after 8 epochs: 1.03

# based on "loss after X epochs" it is actually working

Created on 2024-01-16 with reprex v2.0.2

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 31, 2024
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.

2 participants