remove class_weight parameter