quick look at bart

benhamner · Feb 3, 2015 · 9d7853e · 9d7853e
1 parent bf463a5
commit 9d7853e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/sensitivity.jl b/test/sensitivity.jl
@@ -22,3 +22,17 @@ x1_slope = mean(res.data[:X1])/res.feature_ranges[:X1]
 
 @test x1_slope<15.0
 @test x1_slope>7.5
+
+res = sensitivities(data, :Y, bart_options())
+
+for fea=[:X1, :X2, :X3]
+    println(fea, ": ", mean(res.data[fea]), "\tSlope: ", mean(res.data[fea])/res.feature_ranges[fea])
+end
+
+#@test mean(res.data[:X1])>mean(res.data[:X2])
+#@test mean(res.data[:X2])>mean(res.data[:X3])
+
+x1_slope = mean(res.data[:X1])/res.feature_ranges[:X1]
+
+#@test x1_slope<15.0
+#@test x1_slope>7.5