class_weight = 'auto' for model.fit_generator keras

私は非常に不均衡なデータセットを持っており、model.fit_generatorでclass_weight = 'auto'を使用したいと思います。しかし、私がそれを行うと、私のモデルは学習しません：training_acc = 0.65、val_acc = 0.64（エポック1から50まで） class_weight = 'None'に設定すると、training_acc = 0.92となり、 val_acc =エポック時の0.88。class_weight = 'auto' for model.fit_generator keras

他に誰かがこの問題に直面しましたか？手動でクラスの重みに辞書を定義する必要がありますか？ありがとうございます！あなたの重み付けのための適切なクラス名にtrain_generator.classesを使用することができますkerasでmodel.fit_generatorについて

出典

2017-12-08 anasvaf

次に、あなたの簡単なあなたのクラスをマッピングする辞書を作成します。例えばクラス 'オオカミ' 体重30と 'キツネ' 重み '18'

出典

2017-12-08 09:37:54

OK Giannhの感謝を与える

class_weights = {'wolf':30 , 'fox':18}

！ Na eisai kala！ – anasvaf

読んだり読んだりギリシャ語！私からの投票も。 –

class_weight = 'auto' for model.fit_generator keras

答えて

関連する問題