Skip to content

Commit 03e551c

Browse files
committed
Remove mae workaround in tests
1 parent ccfd1d1 commit 03e551c

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

test/strategies/adaptive.jl

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,10 +91,9 @@ for acceleration in (CPU1(), CPUProcesses(), CPUThreads())
9191
baseline_self_tuning_tree = TunedModel(
9292
model=tree,
9393
tuning=RandomSearch(rng=StableRNG(1234)),
94-
# tuning=ParticleSwarm(n_particles=3, rng=rng),
9594
resampling=CV(nfolds=5, rng=StableRNG(8888)),
9695
range=[r1, r2],
97-
measure=(ŷ, y) -> mean(abs.(ŷ .- y)),
96+
measure=mae,
9897
n=15,
9998
acceleration=acceleration
10099
)
@@ -108,7 +107,7 @@ for acceleration in (CPU1(), CPUProcesses(), CPUThreads())
108107
tuning=AdaptiveParticleSwarm(rng=StableRNG(1234)),
109108
resampling=CV(nfolds=5, rng=StableRNG(8888)),
110109
range=[r1, r2],
111-
measure=(ŷ, y) -> mean(abs.(ŷ .- y)),
110+
measure=mae,
112111
n=15,
113112
acceleration=acceleration
114113
)

test/strategies/basic.jl

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,9 @@ for acceleration in (CPU1(), CPUProcesses(), CPUThreads())
3030
baseline_self_tuning_tree = TunedModel(
3131
model=tree,
3232
tuning=RandomSearch(rng=StableRNG(1234)),
33-
# tuning=ParticleSwarm(n_particles=3, rng=rng),
3433
resampling=CV(nfolds=5, rng=StableRNG(8888)),
3534
range=[r1, r2],
36-
measure=(ŷ, y) -> mean(abs.(ŷ .- y)),
35+
measure=mae,
3736
n=15,
3837
acceleration=acceleration
3938
)
@@ -47,7 +46,7 @@ for acceleration in (CPU1(), CPUProcesses(), CPUThreads())
4746
tuning=ParticleSwarm(rng=StableRNG(1234)),
4847
resampling=CV(nfolds=5, rng=StableRNG(8888)),
4948
range=[r1, r2],
50-
measure=(ŷ, y) -> mean(abs.(ŷ .- y)),
49+
measure=mae,
5150
n=15,
5251
acceleration=acceleration
5352
)

0 commit comments

Comments
 (0)