diff --git a/src/models.jl b/src/models.jl index 9080b45..3919c65 100644 --- a/src/models.jl +++ b/src/models.jl @@ -1,8 +1,8 @@ # Load in models and wrap them to hide @load -const lbc = MLJ.@load LinearBinaryClassifier pkg=GLM -const etc = MLJ.@load EvoTreeClassifier pkg=EvoTrees -const rf = MLJ.@load RandomForestClassifier pkg=DecisionTree +const lbc = MLJ.@load LinearBinaryClassifier pkg=GLM verbosity = 0 +const etc = MLJ.@load EvoTreeClassifier pkg=EvoTrees verbosity = 0 +const rf = MLJ.@load RandomForestClassifier pkg=DecisionTree verbosity = 0 linear_model(; kw...) = lbc(; kw...) boosted_regression_tree(; kw...) = etc(; kw...)