estimator使用
一、model_fn函数有5个输入参数features, labels, mode, params, config,并输出一个EstimatorSpec实例;features:input_fn的第一个输出。 labels:input_fn的第二个输出。 mode:操作类型(是训练、预测还是评估),对应tf.estimator.ModeKeys.EVAL/TRAIN/PREDICT。 params:定义Estimator实例时传入的params属性。 config:定义Estimator实例时
原创
2021-03-30 16:08:31 ·
1298 阅读 ·
0 评论