keras 在不平衡数据上的 fit -- class_weight

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])

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 本系列文章面向深度学习研发者,希望通过Image Caption Generation,一个有意思的具体任务,深入...
    imGeek阅读 1,857评论 0 8
  • 黄油相机:这是一个很文艺的p图软件 我主要用它来拍食物~里面也有自带滤镜 还有贴图 ~不说了上图 然后下面是我的栗子:
    Chocoroo阅读 203评论 0 0
  • 关系 在世间行走不得不游走在关系中 常言道 己所不欲 勿施于人 实际状况确是 因人类强大的自私基因的因素 强迫他人...
    玥烨阅读 252评论 0 0
  • “无人与我立黄昏,无人问我粥可温。” 一个人久了,是不是就习惯了,甚至于遗忘了同他人相伴的滋味? 经历过许多的纷纷...
    HYTing阅读 450评论 0 4