fastai 表格模型训练但找不到分类映射

Question

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

要获取该地图，我想从 Categorify 转换类中获取 self.categories 字典。无论如何可以从通过调用 TabularList.from_df 获得的数据变量中获取它？ 或者也许有人可以告诉我获得这张地图的更好方法。我知道输入 df 到 TabularList.from_df() 不是它，因为行数是错误的。很可能是因为 df 被分成训练子集和有效子集。但是没有简单的方法来获取 TabularList 的火车部分来仅检查火车部分。

很奇怪，我找不到任何显示这一点的代码示例。没有其他人关心将实体嵌入值映射回其原始分类值吗？

Answer 1

我找到了。
它在data.train_ds.inner_df。