python數據處理——樣本數據類別不平衡

keras已經在新版本中加入了 class_weight = 'auto'。
設置了這個參數後,keras會自動設置class weight讓每類的sample對損失的貢獻相等。例子如下:

clf.fit([X_head_train,X_body_train], y_train_embedding, epochs=10, batch_size=128, class_weight = 'auto', validation_data= [[X_head_validate, X_body_validate], y_validate_embedding], callbacks = [tsb])

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章