update parameter value for sklearn 0.17

sklearn 0.17 changes the class_weight argument value 'auto' to 'balanced'.
tmaila · Nov 13, 2015 · 8b79cb9 · 8b79cb9
1 parent dfa6953
commit 8b79cb9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pca/eigenfaces.py b/pca/eigenfaces.py
@@ -91,7 +91,8 @@
          'C': [1e3, 5e3, 1e4, 5e4, 1e5],
           'gamma': [0.0001, 0.0005, 0.001, 0.005, 0.01, 0.1],
           }
-clf = GridSearchCV(SVC(kernel='rbf', class_weight='auto'), param_grid)
+# for sklearn version 0.16 or prior, the class_weight parameter value is 'auto'
+clf = GridSearchCV(SVC(kernel='rbf', class_weight='balanced'), param_grid)
 clf = clf.fit(X_train_pca, y_train)
 print "done in %0.3fs" % (time() - t0)
 print "Best estimator found by grid search:"