How can I solve a multi-label classification problem (not multi-class) in datarobot?
In scikitlearn for example you can pass a vector with all the labels, but I don't know how to pass that to the datarobot, which expects a single column like "target ". A classic example would be the categories of a movie, for example, one can be "action, adventure and suspense" at the same time.
Thanks in advance!!
Solved! Go to Solution.
There is really only one way that I know of to do this in DataRobot currently. You must concatenate the multiple targets into one column. So for example instead of having two target columns “action” “thriller” you would have one column “action, thriller”.
I would also be sure to keep the order consistent across rows so you don’t end up with “thriller, action” as well. I would approach this by alphabetizing them.
I hope this helps,
Hi @doyouevendata, I have also seen in forums that multi-label models can be solved that way (several binary models), but in this case I have about 14 labels in total. For now, I solved it with scikitlearn.
Thank you very much
Ok! I was going to note that with AutoML, it is pretty easy to create many models and deployments, and you may find one approach works better than another for this case or that label and end up with more accurate models as a result. This would likely make it easier to track for accuracy as well if you receive actual values as time goes on. Although at some point the number of labels can make this approach a bit more onerous too.