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

Consider the addition of the pre-trained LASSO model engine #1146

Open
mattwarkentin opened this issue Aug 10, 2024 · 0 comments
Open

Consider the addition of the pre-trained LASSO model engine #1146

mattwarkentin opened this issue Aug 10, 2024 · 0 comments

Comments

@mattwarkentin
Copy link
Contributor

mattwarkentin commented Aug 10, 2024

Hi tidymodels team,

I am not sure if you have seen the recent release of the "pre-trained LASSO" which is now available in the package ptLasso. Not sure if it will find a home here in the tidymodels ecosystem, but I just wanted to bring it to your attention and open some discussion about its potential implementation.

The sales-pitch (from them):

Can pretraining help the lasso? Yes!

Here we present a framework for the lasso in which an overall model is fit to a large set of data, and then fine-tuned to a specific task. This latter dataset can be a subset of the original dataset, but does not need to be. Pretraining the lasso has a wide variety of applications, including stratified models, multinomial targets, multi-response models, conditional average treatment estimation and even gradient boosting.

Paper: http://arxiv.org/pdf/2401.12911
Website: https://erincr.github.io/ptLasso/
R package: https://erincr.github.io/ptLasso/
Other helpful docs: https://www.erincraig.me/lasso-pretraining

This potential model engine may suffer from the same issues as the sparse group LASSO which may preclude its inclusion in tidymodels, which we previously discussed here FWIW: #595

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant