Merge pull request #703 from JuliaAI/performance-evaluation

Fix PerformanceEvaluation bug
JuliaAI · Dec 28, 2021 · 0b6c0ab · 0b6c0ab
2 parents 7311060 + 3a39df3
commit 0b6c0ab
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 12 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "MLJBase"
 uuid = "a7f614a8-145f-11e9-1d2a-a57a1082229d"
 authors = ["Anthony D. Blaom <[email protected]>"]
-version = "0.19"
+version = "0.19.1"
 
 [deps]
 CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"

diff --git a/src/resampling.jl b/src/resampling.jl
@@ -1221,19 +1221,17 @@ function evaluate!(mach::Machine, resampling, weights,
         MLJBase.aggregate(per_fold[k], m)
     end
 
-    ret = (
-        measure = measures,
-        measurement = per_measure,
-        operation = operations,
-        per_fold = per_fold,
-        per_observation = per_observation,
-        fitted_params_per_fold = fitted_params_per_fold |> collect,
-        report_per_fold = report_per_fold |> collect,
-        train_test_rows = resampling
+    return PerformanceEvaluation(
+        measures,
+        per_measure,
+        operations,
+        per_fold,
+        per_observation,
+        fitted_params_per_fold |> collect,
+        report_per_fold |> collect,
+        resampling
     )
 
-    return ret
-
 end
 
 # ----------------------------------------------------------------