set class weights in tenosrlfow estimator code example Example: keras unbalanced data class_weight = {0: 1., 1: 50., 2: 2.}