首页 > 解决方案 > 有没有办法将无效字符串映射到张量流中指定的默认数字?

问题描述

例如,假设我正在使用字符串张量strs = ['12', '52', 'apple', '3'](通常可以是任何维度),并且我想使用内置的 tf 操作将其映射到 float32 张量。如果我只是这样做tf.strings.to_number(strs),那么将出现以下错误消息

StringToNumberOp could not correctly convert string: apple [Op:StringToNumber]

因为显然“苹果”不能解释为数字。

我的问题是:有没有办法为无效条目指定一些默认浮点值(比如-1),以使用纯(矢量化)张量流操作转换为(在本例中为“苹果”)。在这种情况下,所需的输出将是[12, 52, -1, 3].

这样做的原因是因为我们希望在 Keras 层中执行所有数据预处理操作,并尽可能多地消除 Python 依赖关系。

标签: stringtensorflow

解决方案


推荐阅读