本文介绍了如何定义tf.estimator的model_fn()和input_fn()，并探讨了params参数的传递方式。通过阅读源码，解析了Estimator在调用input_fn时params的来源，并在bert源码中找到示例。同时，文章讨论了evaluate()和predict()返回结果的数据结构，以及Estimator和EstimatorSpec的关系，但export_saved_model()部分仍需进一步研究。
参考了官方文档，有些细节问题需要阅读tf源码。
- 如何定义模型model_fn()？需要满足什么格式？官方文档写的还算清楚，
def model_fn(features, labels, mode, params)
其中features和labels是input_fn()的输出结果，params等是新建Estimator对象时填入的各种参数，mode在调用train/evaluate/predict时自动确定。
- 如何定义输入input_fn()？bert源码中input_fn(params)是如何传递params的？通过阅读tf源码终于理解了。这是Estimator类中调用input_fn()的源码，
def _call_input_fn
最低0.47元/天 解锁文章
733
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
