How to obtain the category to index mapping in embedding layer


(Jose Roberto Ayala Solares) #1

I trained a tabular model with some categorical variables. I’d like to know what is the mapping that was used to convert the different categories to integer indexes. In fastai v1.0.42 this was done using learner.data.classes, but in v1.0.45 this is no longer the case. Any help is appreciated.


(Jose Roberto Ayala Solares) #2

A friend of mine figured it out. Just type: learner.data.label_list.train.x.classes