功能描述
获得Rec SDK中自定义的TensorFlow算子。
函数原型
| def import_host_pipeline_ops(so_pkg_name: str = LIBASC_OPS_SO) -> ModuleType
|
参数说明
参数名
|
类型
|
可选/必选
|
说明
|
so_pkg_name
|
string
|
可选
|
Rec SDK so包名。长度范围:[1, 100]
|
返回值说明
- 成功:返回一个包含so中Rec SDK定义的TensorFlow算子的Python封装模块。
- 失败:返回“RuntimeError: when unable to load the library or get the python wrappers.”
使用示例
| from mx_rec.util.ops import import_host_pipeline_ops
host_pipeline_ops = import_host_pipeline_ops()
|
set_threshold算子介绍
算子名称
|
set_threshold
|
算子功能
|
更改特征准入阈值
|
参数说明
|
- 第一个入参为tf上下文中上层的tf.Tensor。
- emb_name: List[str]:需要修改准入阈值的特征表。
- ids_name: List[str]:基于兼容性原因遗留,目前无作用。
|
约束说明
|
set_threshold的第一个入参值为“0”,表示修改对应的emb表为特征不累加模式(准入阈值不变,但是特征计数不再累加,使用历史值)。
|