Skip to content

Commit 1e1eabc

Browse files
committed
Add support for score measurements
1 parent b6234a4 commit 1e1eabc

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/tuning.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,8 @@ function MLJTuning.models(
3737
)
3838
n_particles = tuning.n_particles
3939
if !isnothing(history)
40-
pbest!(state, tuning, map(h -> h.measurement[1], last(history, n_particles)))
40+
sig = MLJTuning.signature(first(history).measure)
41+
pbest!(state, tuning, map(h -> sig * h.measurement[1], last(history, n_particles)))
4142
gbest!(state, tuning)
4243
move!(state, tuning)
4344
end

0 commit comments

Comments
 (0)