From 944be8c685c0bfdb4b4d08842e8a1b8f5cc132ff Mon Sep 17 00:00:00 2001 From: Rishikesh Date: Thu, 16 Feb 2017 03:46:46 +0530 Subject: [PATCH 1/2] Modify SVR example for better plot Change coef0 value for better plot of poly kernal. Also improve the support vectors marking in SVR plot. Fixes: #8365 --- examples/svm/plot_svm_regression.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/examples/svm/plot_svm_regression.py b/examples/svm/plot_svm_regression.py index 15a744e2aa8ca..44142df3afd3e 100644 --- a/examples/svm/plot_svm_regression.py +++ b/examples/svm/plot_svm_regression.py @@ -23,9 +23,9 @@ ############################################################################### # Fit regression model -svr_rbf = SVR(kernel='rbf', C=1e3, gamma=0.1) -svr_lin = SVR(kernel='linear', C=1e3) -svr_poly = SVR(kernel='poly', C=1e3, degree=2) +svr_rbf = SVR(kernel='rbf', C=100, gamma=0.1, epsilon=.1) +svr_lin = SVR(kernel='linear', C=100) +svr_poly = SVR(kernel='poly', C=100, degree=3, epsilon=.1, coef0=1) y_rbf = svr_rbf.fit(X, y).predict(X) y_lin = svr_lin.fit(X, y).predict(X) y_poly = svr_poly.fit(X, y).predict(X) @@ -33,7 +33,12 @@ ############################################################################### # look at the results lw = 2 -plt.scatter(X, y, color='darkorange', label='data') +plt.scatter(X[svr_rbf.support_], y[svr_rbf.support_], facecolor="none", edgecolor="k", marker='^', + label='rbf support vectors', s=40) +plt.scatter(X[svr_lin.support_], y[svr_lin.support_], facecolor="none", edgecolor="k", marker='o', + label='lin support vectors', s=40) +plt.scatter(X[svr_poly.support_], y[svr_poly.support_], facecolor="none", edgecolor="k", marker='s', + label='poly support vectors', s=40) plt.hold('on') plt.plot(X, y_rbf, color='navy', lw=lw, label='RBF model') plt.plot(X, y_lin, color='c', lw=lw, label='Linear model') From 73ba352ac2cdbf57833e5eff155999caa5523842 Mon Sep 17 00:00:00 2001 From: Rishikesh Date: Thu, 16 Feb 2017 20:06:19 +0530 Subject: [PATCH 2/2] Modify plot_svm_regression for removing flake8 error Format a code to remove flake8 error from travis build Issue: #8365 --- examples/svm/plot_svm_regression.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/examples/svm/plot_svm_regression.py b/examples/svm/plot_svm_regression.py index 44142df3afd3e..d18c4f4f36db7 100644 --- a/examples/svm/plot_svm_regression.py +++ b/examples/svm/plot_svm_regression.py @@ -33,12 +33,13 @@ ############################################################################### # look at the results lw = 2 -plt.scatter(X[svr_rbf.support_], y[svr_rbf.support_], facecolor="none", edgecolor="k", marker='^', - label='rbf support vectors', s=40) -plt.scatter(X[svr_lin.support_], y[svr_lin.support_], facecolor="none", edgecolor="k", marker='o', - label='lin support vectors', s=40) -plt.scatter(X[svr_poly.support_], y[svr_poly.support_], facecolor="none", edgecolor="k", marker='s', - label='poly support vectors', s=40) +plt.scatter(X[svr_rbf.support_], y[svr_rbf.support_], facecolor="none", + edgecolor="k", marker='8', label='rbf support vectors', s=100) +plt.scatter(X[svr_lin.support_], y[svr_lin.support_], facecolor="none", + edgecolor="k", marker='^', label='linear support vectors', s=100) +plt.scatter(X[svr_poly.support_], y[svr_poly.support_], facecolor="none", + edgecolor="k", marker='s', label='poly support vectors', s=100) + plt.hold('on') plt.plot(X, y_rbf, color='navy', lw=lw, label='RBF model') plt.plot(X, y_lin, color='c', lw=lw, label='Linear model')