add tests for multivariate regression

panlanfeng · Apr 20, 2017 · f776585 · f776585
1 parent 0a1ab78
commit f776585
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/testreg.jl b/test/testreg.jl
@@ -28,7 +28,12 @@ yfit1=npr(x, y, xeval=xeval, reg=locallinear)
 cb=bootstrapCB(x, y, xeval=xeval)
 @test mean(vec(cb[1,:]) .<= yfit1 .<= vec(cb[2,:])) > .8
 
-
+#multivariate regression
+x = rand(Normal(10), 500, 3)
+y = x * ones(3) .+ x.^2 *ones(3)
+yfit1 = localconstant(x, y, xeval=x)
+regfit = x*inv(x'*x)*x'*y
+@test sum(abs2, regfit .- y) > sum(abs2, yfit1.-y)
 
 
 ###Bounded gamma kernel density and regression