Skip to content

Commit 360e828

Browse files
authored
DOC Use default colors (scikit-learn#16808)
1 parent 97a6b08 commit 360e828

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

examples/feature_selection/plot_feature_selection.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,7 @@
6262
scores = -np.log10(selector.pvalues_)
6363
scores /= scores.max()
6464
plt.bar(X_indices - .45, scores, width=.2,
65-
label=r'Univariate score ($-Log(p_{value})$)', color='darkorange',
66-
edgecolor='black')
65+
label=r'Univariate score ($-Log(p_{value})$)')
6766

6867
# #############################################################################
6968
# Compare to the weights of an SVM
@@ -75,8 +74,7 @@
7574
svm_weights = np.abs(clf[-1].coef_).sum(axis=0)
7675
svm_weights /= svm_weights.sum()
7776

78-
plt.bar(X_indices - .25, svm_weights, width=.2, label='SVM weight',
79-
color='navy', edgecolor='black')
77+
plt.bar(X_indices - .25, svm_weights, width=.2, label='SVM weight')
8078

8179
clf_selected = make_pipeline(
8280
SelectKBest(f_classif, k=4), MinMaxScaler(), LinearSVC()
@@ -89,8 +87,7 @@
8987
svm_weights_selected /= svm_weights_selected.sum()
9088

9189
plt.bar(X_indices[selector.get_support()] - .05, svm_weights_selected,
92-
width=.2, label='SVM weights after selection', color='c',
93-
edgecolor='black')
90+
width=.2, label='SVM weights after selection')
9491

9592

9693
plt.title("Comparing feature selection")

0 commit comments

Comments
 (0)