Skip to content

Commit

Permalink
update tuning test
Browse files Browse the repository at this point in the history
  • Loading branch information
ablaom committed Aug 4, 2019
1 parent ceb18e4 commit 2aea48c
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion test/tuning.jl
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@ using MLJ
# using UnicodePlots
using MLJBase
using CSV
import Random.seed!
seed!(1234)

x1 = rand(100);
x2 = rand(100);
Expand Down Expand Up @@ -46,8 +48,9 @@ measurements = tuned.report.measurements
# should be all different:
@test length(unique(measurements)) == length(measurements)

# the following should only fail in extremely rare event:
@test length(b.transformer.features) == 3
@test abs(b.model.lambda - 0.027825) < 1e-6


# get the training error of the tuned_model:
e = rms(y, predict(tuned, X))
Expand Down

0 comments on commit 2aea48c

Please sign in to comment.