Skip to content

Commit f868e94

Browse files
authored
fix default vaulues for mlp(engine = "brulee") (#1020)
* fix defaults * update news
1 parent b462f76 commit f868e94

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

NEWS.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
* Improved errors in cases where the outcome column is mis-specified. (#1003)
44

5+
* Documentation fixed for `mlp(engine = "brulee")`: the default values for `learn_rate` and `epochs` were swapped (#1018).
6+
57
# parsnip 1.1.1
68

79
* Fixed bug where prediction on rank deficient `lm()` models produced `.pred_res` instead of `.pred`. (#985)

man/rmd/mlp_brulee.Rmd

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
```{r brulee-param-info, echo = FALSE}
99
defaults <-
1010
tibble::tibble(parsnip = c("hidden_units", "penalty", "dropout", "epochs", "learn_rate", "activation", "mixture"),
11-
default = c("3L", "0.0", "0.0", "0.01", "100L", "'relu'", "0.0"))
11+
default = c("3L", "0.0", "0.0", "100L", "0.01", "'relu'", "0.0"))
1212
1313
param <-
1414
mlp() %>%

man/rmd/mlp_brulee.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,11 @@ This model has 7 tuning parameters:
1515

1616
- `mixture`: Proportion of Lasso Penalty (type: double, default: 0.0)
1717

18-
- `epochs`: # Epochs (type: integer, default: 0.01)
18+
- `epochs`: # Epochs (type: integer, default: 100L)
1919

2020
- `dropout`: Dropout Rate (type: double, default: 0.0)
2121

22-
- `learn_rate`: Learning Rate (type: double, default: 100L)
22+
- `learn_rate`: Learning Rate (type: double, default: 0.01)
2323

2424
- `activation`: Activation Function (type: character, default: 'relu')
2525

0 commit comments

Comments
 (0)