How to get class assignment for columns for predict_proba?

[nadiiach]

predict_proba returns probabilities for corresponding classes and user needs to know the class correspondence.
In other frameworks this information is accessible through .classes_ instance variable of estimator.
Is there anything like this for DeepTables models? I am currently working with DCN.

[jackguagua]

@nadiiach nice suggestion! DT will support it soon.

[jackguagua]

Now, you can get classes information like this:

dt.fit(X,y)
dt.classes_
dt.num_classes
dt.pos_label  #only for binary classification