diff --git a/test/runtests.jl b/test/runtests.jl index 19eadda..78b3307 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -153,6 +153,9 @@ end fit!(mach, verbosity=0) yhat = predict(mach, Xtable) + imps = feature_importances(mach) + @test Set(string.([imp[1] for imp ∈ imps])) == Set(["x1", "x2", "x3"]) + # serialize: io = IOBuffer() MLJBase.save(io, mach)