读取当前step已保存的稀疏表checkpoint,以key-value形式保存为numpy文件,其中key为查询ID,value为嵌入层表示。以下情况可支持使用export接口导出稀疏表数据:
1 | def export(table_list=None): |
参数名 |
类型 |
可选/必选 |
说明 |
---|---|---|---|
table_list |
list[str] |
可选 |
需要导出的稀疏表列表,不传参默认导出所有稀疏表。取值范围:[1, 2^31-1] |
1 2 3 | from mx_rec.saver.sparse import export table_list=["sparse_embedding_table"] export(table_list=table_list) |
接口调用流程及示例,请参见模型迁移与训练。