Pytorch fastai表格模型经过训练,但无法找到分类映射

Pytorch fastai表格模型经过训练,但无法找到分类映射,pytorch,categorical-data,embedding,fast-ai,Pytorch,Categorical Data,Embedding,Fast Ai,在使用fastai的表格模型对包含大量分类数据的数据集进行训练之后,我希望读取实体嵌入并使用它映射到原始数据值。 我可以看到嵌入的权重。输入的数量似乎与任何内容都不匹配,但可能是基于序列中唯一的分类值 为了获得该映射,我想从Categorify转换类中获取self.categories字典。通过调用tablerlist.from_df获得的数据变量是否可以得到它? 或者有人能告诉我一个更好的方法来得到这张地图。我知道表格列表中的输入df。from_df()不是,因为行数不对。最有可能是因为df被

在使用fastai的表格模型对包含大量分类数据的数据集进行训练之后,我希望读取实体嵌入并使用它映射到原始数据值。 我可以看到嵌入的权重。输入的数量似乎与任何内容都不匹配,但可能是基于序列中唯一的分类值

为了获得该映射,我想从Categorify转换类中获取self.categories字典。通过调用tablerlist.from_df获得的数据变量是否可以得到它? 或者有人能告诉我一个更好的方法来得到这张地图。我知道表格列表中的输入df。from_df()不是,因为行数不对。最有可能是因为df被划分为序列和有效子集。但是,没有简单的方法可以获取列表中的火车部分来检查火车部分

奇怪的是,我找不到任何代码示例来说明这一点。难道其他人不想将实体嵌入值映射回其原始分类值吗?

我找到了它。
它位于
data.train\ds.internal\u df