Improve dataloading for regression
The following discussion from !118 (merged) should be addressed:
-
@svanstro started a discussion: It would be slightly more robust to check if
"regression" in name
, but still not ideal. Really we would have access to the full task object here to check whether it is a regression task or classification task. I will have a think about how we can do that