Multilabel modeling

Multilabel modeling

Hello ,
I wanted to try Multilabel modeling on datarobt based on the doc :

I reproduced the test dataset with the code from the doc but datarobot does not detect the Target as Multilabel.



the code to produce the test dataset :

import json

import pandas as pd

multicategorical_values = [["A", "B"], ["A"], ["A", "C"], ["B", "C"] ,["B", "C"], ["B", "C"] , ["B", "C"] ,["B", "C"] ,["B", "C"] ,["B", "C"]

                          ,["B", "C"],["B", "C"],["B", "C"],["B", "C"],["B", "C"],["B", "C"],["B", "C"],["B", "C"],["B", "C"],["B", "C"]


df = pd.DataFrame(


        "numeric_feature": [1, 2, 3, 4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20],

        "multicategorical_feature": multicategorical_values,




df["multicategorical_feature"] = df["multicategorical_feature"].apply(json.dumps)

df.to_csv("dataset_test.csv", index=False)


Thank you for your time!


Labels (1)
0 Kudos
1 Reply
DataRobot Employee
DataRobot Employee


Just had someone test the code and it seems to work. Can you check that you have the "Enable Multilabel Projects and Multicategorical Feature Types" Feature Flag enabled?

0 Kudos