2017-05-22 2 views
0

Kerasの不平衡クラスをトレーニングするには、sample_weightsパラメータを設定する必要があります。Kerasの適合モデルまたは評価モデルの両方でsample_weightsを使用できません

私は重みのリストを使用しようとする場合、それはエラーを与える: はAttributeError:私は1次元配列を使用しようとすると「リスト」オブジェクトが属性「形状」を持っていない

は、それがエラーを与える: ValueError:shape(180、17)の入力のshape(17、)を持つs​​ample_weight配列が見つかりました。 sample_weightはブロードキャストできません。

model.evaluate(X, Y, batch_size=50, sample_weight=weights) 

この日の終わりには、私はこのパラメータを渡そうとしましたが、何らかのエラーが発生しました。

アイデア?

答えて

0

体重は(180,)とする必要があります。トレーニングデータの1サンプルあたり1つの重量。

0

正しいパラメータはclass_weightで、重みの辞書を受け取ります。

weights = dict([0:1, 1:2, 2:1, 3:5]) 
model.evaluate(X, Y, batch_size=50, class_weight=weights) 
関連する問題