标签: machine-learning tensorflow
在张量流estimator/estimator.py,函数fit,partial_fit,predict等中,它表示在使用input_fn时,batch_size必须没有。 那么,在哪里设置batch_size。在我的代码中,需要input_fn,因为我用它来从原始数据中选择一些特征。 感谢。
estimator/estimator.py
fit
partial_fit
predict
input_fn
batch_size