I am trying to build a Linear Regression model using a not so big dataset. I'm more comfortable doing classification and I am not really an expert in regression.

In classification, I was used to build train/test sets containing, more or less, the same number of elements for each class. I was wondering whether there is something similar for regression: my dataset is composed of around 100 rows (elements, like animals), but some elements are much more representative than others. Is it possible to create (for instance, with SKlearn) train/test sets which are balanced? I was thinking about the possibility to pass a vector containing the labels of each element to the function doing the split.


Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.