Skip to content

Commit b768e06

Browse files
authored
If verbosity > 0, print the name of the selected solver when fit! is called
1 parent 0e19298 commit b768e06

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/mlj/interface.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ function MMI.fit(m::Union{REG_MODELS...}, verb::Int, X, y)
3636
features = (sch === nothing) ? nothing : sch.names
3737
reg = glr(m)
3838
solver = m.solver === nothing ? _solver(reg, size(Xmatrix)) : m.solver
39+
verb > 0 && @info "Solver: $(solver)"
3940
# get the parameters
4041
θ = fit(reg, Xmatrix, y; solver=solver)
4142
# return
@@ -72,6 +73,7 @@ function MMI.fit(m::Union{CLF_MODELS...}, verb::Int, X, y)
7273
# NOTE: here the number of classes is either 0 or > 2
7374
clf = glr(m, nclasses)
7475
solver = m.solver === nothing ? _solver(clf, size(Xmatrix)) : m.solver
76+
verb > 0 && @info "Solver: $(solver)"
7577
# get the parameters
7678
θ = fit(clf, Xmatrix, yplain, solver=solver)
7779
# return

0 commit comments

Comments
 (0)