imodels icon indicating copy to clipboard operation
imodels copied to clipboard

support categorical input features?

Open Yannahhh opened this issue 2 years ago • 7 comments

It seems that this package does not support categorical variables? Right?

Yannahhh avatar Jan 08 '22 11:01 Yannahhh

That is correct, we are working on this but for now if you want to use categorical variables you'll have to first one-hot encode them.

csinva avatar Jan 08 '22 16:01 csinva

I was trying BoostedRulesClassifier for categorical labels. It does not work even after one hot. I assume this only supports Binary Classification. One can use one hot but only for input features.

paularindam avatar Feb 14 '22 07:02 paularindam

Good point! You are correct my above comment was about categorial feature variables which an be used with one-hot encoding.

Most of the models here only support binary classification (see this table) for details.

csinva avatar Feb 14 '22 21:02 csinva

Is there anyone which supports multi-class classification? I can see binary classification and regression. Do you plan to add multi-class support ?

paularindam avatar Feb 14 '22 21:02 paularindam

We do plan to add it although it may not be for another month - RuleFit, SLIM, FIGS, and HSTree (among others) should easily support multi-class, but we haven't been testing it so there may be an issue when using them for the multi-class setting.

csinva avatar Feb 14 '22 21:02 csinva

Thanks for the prompt response.

paularindam avatar Feb 14 '22 21:02 paularindam

@paularindam FIGS just received categorical support in https://github.com/csinva/imodels/pull/151

mepland avatar Dec 29 '22 17:12 mepland