OrphanLookupKeySlicerHook
该Hook用于支持稀疏表查询时传入向上无法找到Dataset的孤儿Key类型,主要用于拓展自动改图模式下的稀疏表查询功能。如果没有找到目标类型的算子,不会执行切分操作。
 该Hook的使用场景是NPUEstimator模式,启用自动改图功能。该Hook的调用时机需要在自动改图的GraphModifierHook之前。
使用示例
from mx_rec.graph import OrphanLookupKeySlicerHook, GraphModifierHook
def input_fn():
    """
    用户自定义Estimator输入函数。
    """
orphan_slicer_hook = OrphanLookupKeySlicerHook()
modifier_hook = GraphModifierHook(modify_graph=params.modify_graph)
hooks_list = [orphan_slicer_hook, modifier_hook]
est = NPUEstimator(...)
est.train(input_fn=lambda: input_fn, hooks=npu_hooks_append(hooks_list))
父主题: 自动改图