在忙碌的训练中如何从model.fit()获取输出/日志

时间:2018-08-18 16:00:48

标签: tensorflow.js

我是Tensorflow.js的新手。

来自Keres,我习惯于在训练过程中打印出这样的默认日志:

Epoch 1/2
320/320 [==============================] - 7s 21ms/step - loss: 5.6595
Epoch 2/2
320/320 [==============================] - 5s 15ms/step - loss: 5.5646

但是,当我在Tensorflow.js中运行await model.fit()时,我的控制台只是挂起而没有任何输出。有没有可以传递给model.fit()的设置,以便可以在培训期间将某种基本日志输出到控制台?

1 个答案:

答案 0 :(得分:1)

您可以将回调传递到.fit(),该回调在不同事件(df['group_x1']=np.arange(len(df))//5 df Out[170]: x1 x2 group_x1 0 0 20 0 1 1 20 0 2 2 20 0 3 3 20 0 4 4 20 0 5 5 20 1 6 6 20 1 7 7 20 1 8 8 20 1 9 9 20 1 10 10 20 2 onTrainBeginonTrainEndonEpochBeginonEpochEndonBatchBegin

示例:

onBatchEnd