search

jukaje / mlr-mini

2 stars 0 forks source link

Dataset() check if regression or classification task #6

Open sophiafrei opened 1 year ago

sophiafrei commented 1 year ago

Hi,

i think it would make sense to ask in the Dataset()-function if the target variable is factorial or not and then assign depending on this class "DatasetClassification" or "DatasetRegression". What do you think?

sophiafrei commented 1 year ago

Because right now the class of an iris task with target "species" is DatasetRegression:

ir <- Dataset(iris, target = "species", type = "classification") class(ir) [1] "DatasetRegression" "Dataset"