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