Python 将样本/类别权重应用于拟合生成器时出错_Python_Tensorflow

Python 将样本/类别权重应用于拟合生成器时出错

python tensorflow

Python 将样本/类别权重应用于拟合生成器时出错,python,tensorflow,Python,Tensorflow,我正在使用tf.keras.Model fit_生成器（）将成批数据提供给模型。根据TensorFlow文档，拟合生成器应该能够接受大小为2（输入，目标）或3（输入，目标，样本权重）的元组。我们有大小2的工作，但我们有不平衡的类，所以我已经确定了样本权重。当fit生成器返回一个大小为3的元组时，我得到一个错误： tensorflow.python.framework.errors\u impl.InvalidArgumentError:无法压缩dim[0]，维度应为1，获得[batch\u si

我正在使用tf.keras.Model fit_生成器（）将成批数据提供给模型。根据TensorFlow文档，拟合生成器应该能够接受大小为2（输入，目标）或3（输入，目标，样本权重）的元组。我们有大小2的工作，但我们有不平衡的类，所以我已经确定了样本权重。当fit生成器返回一个大小为3的元组时，我得到一个错误： tensorflow.python.framework.errors\u impl.InvalidArgumentError:无法压缩dim[0]，维度应为1，获得[batch\u size]

我使用的是tensorflow 1.12

损失函数是tf.Loss.softmax\u cross\u entropy

您可以添加一些代码和数据集吗？