class_weights = {0: 1, 1: 10}