changed state_weight back to class_weight, this will have to be a larger change, if we do it