do svm and ranger Pt.6

Techtonique · Jan 18, 2024 · 009e4b1 · 009e4b1
1 parent 4958f88
commit 009e4b1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -29,14 +29,14 @@ using Test
     end
 
     @testset "----- Testing dynrmf (svm) -----" begin
-        val = Ahead.dynrmf(y, h=6, kernel="linear")
+        val = Ahead.dynrmf(y, h=6, method="svm", kernel="linear")
         println(val) 
         @test isapprox(round(val[:residuals][1]), 0)
         @test val[:x] == y
     end
 
     @testset "----- Testing dynrmf (ranger) -----" begin
-        val = Ahead.dynrmf(y, h=6, num_trees=50)
+        val = Ahead.dynrmf(y, h=6, method="randomforest", num_trees=50)
         println(val) 
         @test isapprox(round(val[:residuals][1]), 0)
         @test val[:x] == y