aerdem4 / lofo-importance

Leave One Feature Out Importance
MIT License
810 stars 83 forks source link

Performing feature selection in multi targeted regression dataset #61

Open sametsoekel opened 2 months ago

sametsoekel commented 2 months ago

Hi,

I would like to apply lofo in my multi targeted regression dataset, here is a reprex of my data;


from sklearn.datasets import make_regression

X,y = make_regression(n_samples=int(1e3),n_features=5,n_targets=4)

some models (like CatBoost) allow multi targets without changing any parameter.

Since lofo.Dataset takes target= argument as string, I couldn't perform feature selection with lofo-importance.

Can you add this compatibility ?

Thanks

aerdem4 commented 1 month ago

Can you please share the code for me to be able to reproduce the issue?