在培訓期間更改keras class_weight

Question

我想利用Keras中的fit函數的參數class_weight來處理不平衡的數據集。

關鍵在於我想通過基於正在運行的時代的函數來指定class_weights。

我雖然要使用回調函數，但我無法從那裏更改參數class_weight。任何幫助？

Answer 1

此時不幸的是無法在回調函數中更改class_weight參數。該參數不會保存爲模型中的實例變量，因此您將無法訪問它。

但是，如果您使用model.train_on_batch，則可以調整每個批次的類別權重參數。您將能夠從批號中計算新紀元號碼。但是，這隻適用於這種特殊的適合方法。因此，在將批數據傳遞給fit函數之前，您需要創建批生成器並修改類權重參數。