据我了解Estimator
API,方法train
会对给定数据进行训练,直到input_fn
函数引发异常或直到达到步数为止。因此,如果我想循环直到估算器已经收敛(对于它的某些定义),我需要自己编写循环并测试我的标准。对于估算工具e
:
prevloss = 999999999999
while True:
e.train(input_fn)
loss = e.evaluate(input_fn)['loss']
if abs(prevloss - loss) < 1e-4:
break
prevloss = loss
但有一些事情对我来说仍然模糊不清。