- Notifications
You must be signed in to change notification settings - Fork9
High-Level Modeling Functions with 'torch'
License
Unknown, MIT licenses found
Licenses found
tidymodels/brulee
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
The Rbrulee package contains several basic modeling functions thatuse thetorch package infrastructure, such as:
You can install the released version of brulee fromCRAN with:
install.packages("brulee")And the development version fromGitHub with:
# install.packages("pak")pak::pak("tidymodels/brulee")
brulee has formula, x/y, and recipe user interfaces for each function.For example:
library(brulee)library(recipes)library(yardstick)data(bivariate,package="modeldata")set.seed(20)nn_log_biv<- brulee_mlp(Class~ log(A)+ log(B),data=bivariate_train,epochs=150,hidden_units=3)# We use the tidymodels semantics to always return a tibble when predictingpredict(nn_log_biv,bivariate_test,type="prob")|> bind_cols(bivariate_test)|> roc_auc(Class,.pred_One)#> # A tibble: 1 × 3#> .metric .estimator .estimate#> <chr> <chr> <dbl>#> 1 roc_auc binary 0.840
A recipe can also be used if the data require some sort of preprocessing(e.g., indicator variables, transformations, or standardization):
library(recipes)rec<- recipe(Class~.,data=bivariate_train)|> step_YeoJohnson(all_numeric_predictors())|> step_normalize(all_numeric_predictors())set.seed(20)nn_rec_biv<- brulee_mlp(rec,data=bivariate_train,epochs=150,hidden_units=3)# A little betterpredict(nn_rec_biv,bivariate_test,type="prob")|> bind_cols(bivariate_test)|> roc_auc(Class,.pred_One)#> # A tibble: 1 × 3#> .metric .estimator .estimate#> <chr> <chr> <dbl>#> 1 roc_auc binary 0.862
Please note that the brulee project is released with aContributor CodeofConduct.By contributing to this project, you agree to abide by its terms.
About
High-Level Modeling Functions with 'torch'
Resources
License
Unknown, MIT licenses found
Licenses found
Code of conduct
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Packages0
Contributors13
Uh oh!
There was an error while loading.Please reload this page.
