cancel
Showing results for 
Search instead for 
Did you mean: 

Multilabel modeling

Multilabel modeling

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

https://docs.datarobot.com/en/docs/modeling/special-workflows/multilabel.html#multilabel-modeling

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

 

lilia_0-1704968045155.png

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!
Sincerely

 

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

Hello,

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