Fatal编程技术网
  • python/
  • Python 等级重量='；自动'；适用于fit_型发电机keras

Python 等级重量='；自动'；适用于fit_型发电机keras

python machine-learning neural-network deep-learning keras

Python 等级重量='；自动'；适用于fit_型发电机keras,python,machine-learning,neural-network,deep-learning,keras,Python,Machine Learning,Neural Network,Deep Learning,Keras,我有一个高度不平衡的数据集，我想在模型.fit\u生成器中使用class\u weight='auto'。然而，当我这样做时，我看到我的模型没有学习：training\u acc=0.65和val\u acc=0.64从第1纪元开始一直到第50纪元。 如果我设置class\u weight='None'则模型开始学习：training\u acc=0.92，val\u acc=0.88，在第50个历元 还有其他人面临这个问题吗？我必须手动为我的类权重定义字典吗？对于模型。在keras中安装生成器

我有一个高度不平衡的数据集，我想在
模型.fit\u生成器中使用
class\u weight='auto'
。然而，当我这样做时，我看到我的模型没有学习：
training\u acc=0.65
和
val\u acc=0.64
从第1纪元开始一直到第50纪元。
如果我设置
class\u weight='None'
则模型开始学习：
training\u acc=0.92，val\u acc=0.88，在第50个历元


还有其他人面临这个问题吗？我必须手动为我的类权重定义字典吗？
对于
模型。在keras中安装生成器可以使用
train\u generator.classes
为您的权重定义正确的类名

然后创建一个映射类的字典，例如


class_权重={'wolf'：30，'fox'：18}


这使得“狼”类的权重为30，“狐狸”类的权重为18