fix bug with test, train split

JuliaAI · Dec 18, 2024 · a8902d3 · a8902d3
1 parent fcebd20
commit a8902d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "MLJTestInterface"
 uuid = "72560011-54dd-4dc2-94f3-c5de45b75ecd"
 authors = ["Anthony D. Blaom <[email protected]>"]
-version = "0.2.7"
+version = "0.2.8"
 
 [deps]
 MLJBase = "a7f614a8-145f-11e9-1d2a-a57a1082229d"

diff --git a/src/attemptors.jl b/src/attemptors.jl
@@ -93,7 +93,7 @@ function operations(fitted_machine, data...; throw=false, verbosity=1)
         model = fitted_machine.model
         operations = String[]
         methods = MLJBase.implemented_methods(fitted_machine.model)
-        _, test = MLJBase.partition(1:MLJBase.nrows(first(data)), 0.99)
+        _, test = MLJBase.partition(1:MLJBase.nrows(first(data)), 0.01)
         if :predict in methods
             predict(fitted_machine, first(data))
             model isa Static || predict(fitted_machine, rows=test)