From 3a8b08efc567b25aa3eec3fcc5d88a3313dae1fb Mon Sep 17 00:00:00 2001 From: Thibaut Lienart Date: Fri, 8 May 2020 10:46:26 +0200 Subject: [PATCH] closes #291 --- src/data/datasets_synthetic.jl | 3 ++- test/data/datasets_synthetic.jl | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/data/datasets_synthetic.jl b/src/data/datasets_synthetic.jl index cf927160..8d24b351 100644 --- a/src/data/datasets_synthetic.jl +++ b/src/data/datasets_synthetic.jl @@ -355,5 +355,6 @@ function make_regression(n::Int=100, p::Int=2; intercept::Bool=true, end end - return finalize_Xy(X, y, false, as_table, eltype, rng; clf=binary) + return finalize_Xy(X[:,1:end-Int(intercept)], y, false, + as_table, eltype, rng; clf=binary) end diff --git a/test/data/datasets_synthetic.jl b/test/data/datasets_synthetic.jl index d814899b..934e8924 100644 --- a/test/data/datasets_synthetic.jl +++ b/test/data/datasets_synthetic.jl @@ -82,7 +82,7 @@ end n, p = 100, 5 X, y = make_regression(n, p) Xm = MLJBase.matrix(X) - @test size(Xm) == (n, p+1) + @test size(Xm) == (n, p) @test length(y) == n # specific arguments end