Python 如何使用适用于tf1.15的DNNLinearCombinedClassifier向标签添加随机噪声？

目前，我正在使用tf1.15中的宽深分类器（DNNLinearCombinedClassifier），并定义了以下函数来向标签添加随机噪声：

def add_random_noise(labels):
    labels = tf.dtypes.cast(labels, tf.float32)
    rnd_noise = tf.random.uniform(tf.shape(labels))
    return tf.add(labels, tf.math.multiply_no_nan(0.05, rnd_noise))

然后使用

map

将此函数应用于数据集，并仅应用于目标列。但是，当我训练我的模型时，它会生成以下输出：

(0) Invalid argument: assertion failed: [Labels must <= n_classes - 1] [Condition x <= y did not hold element-wise:] [x (head/labels:0) = ] [[0.00486446638][0.0133116841][0.0143840136]...] [y (head/assert_range/Const:0) = ] [1]

（0）无效参数：断言失败：[标签必须