Update

lanl-ansi · Oct 18, 2024 · 22c461b · 22c461b
1 parent b20f4c2
commit 22c461b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_DecisionTree.jl b/test/test_DecisionTree.jl
@@ -47,10 +47,10 @@ function test_DecisionTree()
     return
 end
 
-function test_DecisionTree()
+function test_DecisionTreeClassifier()
     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.DecisionTreeClassifier(; max_depth = 3)
     DecisionTree.fit!(ml_model, features, labels)