Merge pull request #37 from DilumAluthge/patch-1

Fix ScikitLearn API regression example
JuliaML · Jul 12, 2018 · 37fbf12 · 37fbf12
2 parents 5b3d530 + 29274f4
commit 37fbf12
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -61,6 +61,9 @@ yp = predict(model, instances[2:2:end, :])
 #epsilon-regression
 whiteside = RDatasets.dataset("MASS", "whiteside")
 X = Array(whiteside[:Gas])
+if typeof(X) <: AbstractVector
+    X = reshape(X, (length(X),1))
+end
 y = Array(whiteside[:Temp])
 svrmod = fit!(EpsilonSVR(cost = 10., gamma = 1.), X, y)
 yp = predict(svrmod, X)