get_asc_insert_func

功能描述

获取数据预处理函数。

函数原型

def get_asc_insert_func(tgt_key_specs=None, args_index_list=None, table_names=None, **kwargs)

参数说明

参数名	类型	可选/必选	说明
tgt_key_specs	FeatureSpec list[FeatureSpec]	必选，可使用两种方式传入，具体请参见说明。	特征对象或特征对象列表或者特征对象元组，默认值为“None”。
args_index_list	list		参数索引列表，默认值为“None”。
table_names	list		表名称列表，默认值为“None”。

接口参数可选择以下其中一种方式传入。

仅传入“tgt_key_specs”。
传入“args_index_list”和“table_names”。

**kwargs参数说明

参数名	类型	可选/必选	说明
is_training	bool	可选	是否为训练模式，默认值为“True”。取值范围： True：训练模式 False：评估或预测模式
dump_graph	bool	可选	是否保存模型图，默认值为“False”。取值范围： True：保存模型图 False：不保存模型图

参数名

类型

可选/必选

说明

is_training

bool

可选

是否为训练模式，默认值为“True”。

取值范围：

True：训练模式
False：评估或预测模式

dump_graph

bool

可选

是否保存模型图，默认值为“False”。

取值范围：

True：保存模型图
False：不保存模型图

**kwargs参数中的“is_training”和“dump_graph”作为内部使用参数，不建议用户通过kwargs传递这两个参数。
如果通过kwargs传递其他未说明参数，则Rec SDK内部不会使用到该参数。

返回值说明

成功：数据预处理函数。
失败：抛出异常。

使用示例

import tensorflow as tf
from mx_rec.core.asc.helper import get_asc_insert_func

dataset = tf.data.TFRecordDataset(data_path) # data_path为数据集路径
dataset = dataset.map(get_asc_insert_func(tgt_key_specs=feature_spec_list, is_training=True)) # feature_spec_list中元素为FeatureSpec对象

参考资源

接口调用流程及示例，参见模型迁移与训练。

父主题： 数据接口