OrphanLookupKeySlicerHook

该Hook用于支持稀疏表查询时传入向上无法找到Dataset的孤儿Key类型,主要用于拓展自动改图模式下的稀疏表查询功能。如果没有找到目标类型的算子,不会执行切分操作。

该Hook的使用场景是NPUEstimator模式,启用自动改图功能。该Hook的调用时机需要在自动改图的GraphModifierHook之前。

使用示例

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
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))