get_asc_insert_func
功能描述
获取数据预处理函数。
函数原型
def get_asc_insert_func(tgt_key_specs=None, args_index_list=None, table_names=None, **kwargs)
参数说明
参数名 |
类型 |
可选/必选 |
说明 |
|---|---|---|---|
tgt_key_specs |
|
必选，可使用两种方式传入，具体请参见说明。 |
特征对象或特征对象列表或者特征对象元组，默认值为“None”。 |
args_index_list |
list |
参数索引列表，默认值为“None”。 |
|
table_names |
list |
表名称列表，默认值为“None”。 |
接口参数可选择以下其中一种方式传入。
- 仅传入“tgt_key_specs”。
- 传入“args_index_list”和“table_names”。
**kwargs参数说明
参数名 |
类型 |
可选/必选 |
说明 |
|---|---|---|---|
is_training |
bool |
可选 |
是否为训练模式，默认值为“True”。 取值范围：
|
dump_graph |
bool |
可选 |
是否保存模型图，默认值为“False”。 取值范围：
|
- **kwargs参数中的“is_training”和“dump_graph”作为内部使用参数，不建议用户通过kwargs传递这两个参数。
- 如果通过kwargs传递其他未说明参数，则mxRec内部不会使用到该参数。
返回值说明
- 成功：数据预处理函数。
- 失败：抛出异常。
使用示例
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对象
参考资源
接口调用流程及示例，参见模型训练。
父主题： 数据接口