Update

lanl-ansi · Oct 18, 2024 · 95264fc · 95264fc
1 parent b101fbc
commit 95264fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_DecisionTree.jl b/test/test_DecisionTree.jl
@@ -26,7 +26,7 @@ end
 function test_DecisionTree()
     truth(x::Vector) = x[1] <= 0.5 ? -2 : (x[2] <= 0.3 ? 3 : 4)
     rng = Random.MersenneTwister(1234)
-    features = rand(rng, 10, 2)
+    features = rand(rng, 100, 2)
     labels = truth.(Vector.(eachrow(features)))
     ml_model = DecisionTree.build_tree(labels, features)
     model = Model(HiGHS.Optimizer)