tensorflow change weights unbalanced dataset code example

Example: keras unbalanced data

class_weight = {0: 1.,
                1: 50.,
                2: 2.}