小编tud*_*r.a的帖子

>     import tensorflow as tf
>     
>     class MyMetric(tf.keras.callbacks.Callback):
>        def on_epoch_end(self,epoch,logs={}):
>            # how to access X_train and X_val here
> 
>     ...
>     model.fit(X_train,y_train,batch_size=32,epochs=10,validation_data=(X_val,y_val),shuffle=True,callbacks=[MyMetric()]

我正在尝试使用回调在tensorflow 2.0 中实现自定义指标。在该on_epoch_end方法中，我需要访问提供给拟合方法的训练和验证数据（整个样本，而不是批次）。有什么办法可以做到这一点吗？谢谢！