fixed error message for invalid class_weight values

neurodata · Jun 1, 2016 · 5a333cd · 5a333cd
1 parent 1f76574
commit 5a333cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sklearn/utils/class_weight.py b/sklearn/utils/class_weight.py
@@ -71,7 +71,7 @@ def compute_class_weight(class_weight, classes, y):
  # user-defined dictionary
  weight = np.ones(classes.shape[0], dtype=np.float64, order='C')
  if not isinstance(class_weight, dict):
- raise ValueError("class_weight must be dict, 'auto', or None,"
+ raise ValueError("class_weight must be dict, 'balanced', or None,"
  " got: %r" % class_weight)
  for c in class_weight:
  i = np.searchsorted(classes, c)