add support for class_weights