时间戳

首页 > 解决方案 > Tensorflow 对象到一个 numpy 数组

python - Tensorflow 对象到一个 numpy 数组

问题描述

我正在尝试在 Keras 中使用自定义 F1 评分函数，但我的目标变量是一个数组，我不想对它们进行热编码，所以我将sparse_categorical_crossentropy其用作损失。

from keras import backend as K
def f1(y_true, y_pred):
  print(type(y_true))
  return y_true

当我运行上面的代码时，我得到 y_true 和 y_pred 为<class 'tensorflow.python.framework.ops.Tensor'>. 我们如何将 y_true 和 y_pred 转换为 NumPy 数组，以便我可以在它们上实现 sklearn 的 F1 分数功能？

标签： pythonnumpytensorflowkeras

解决方案

使用（正如@learner 在评论中所建议的那样）：

y_true = y_true.numpy()
y_pred = y_pred.numpy()

推荐阅读