我是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()
的设置,以便可以在培训期间将某种基本日志输出到控制台?
答案 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
,onTrainBegin
,onTrainEnd
,onEpochBegin
,onEpochEnd
,onBatchBegin
)
示例:
onBatchEnd