load models silently

tiemvanderdeure · Nov 29, 2023 · 5d1469b · 5d1469b
1 parent 73fb8f8
commit 5d1469b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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...)