search

AutoViML / Auto_ViML

Automatically Build Multiple ML Models with a Single Line of Code. Created by Ram Seshadri. Collaborators Welcome. Permission Granted upon Request.
Apache License 2.0
522 stars 102 forks source link

groups in cross validation? #22

Closed crsegerie closed 11 months ago

crsegerie commented 3 years ago

I would like to use the group params in the cross validation like in sklearn ? Is it possible to do so ?

Otherwise, thank you for your package, this is a very cool work here, especially the visualization part.

rsesha commented 3 years ago

Hi 👍 Thanks for the appreciation. Is this what you are looking for? Link groups: array-like of shape (n_samples,), default=None Group labels for the samples used while splitting the dataset into train/test set. Only used in conjunction with a “Group” cv instance (e.g., GroupKFold).

Since I don't use GroupKFold, but just regular KFold, do you want an input argument that says "groups" and if you send the groups, I can use it with the GroupKFold?

Please let me know and I'll consider it. Thanks Ram

crsegerie commented 3 years ago

yes, this is exactly it!

AutoViML commented 11 months ago

Hi @crsegerie 👍 Can you create a pull request and take a crack at writing some code? I don't have much time and would appreciate the help, Thanks AutoVimal