Skip to content

Commit 33fa2e2

Browse files
committed
change signature of training_losses
1 parent 6ca5250 commit 33fa2e2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/model_api.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ update(m::Model, verbosity, fitresult, cache, data...) =
2626
fit(m, verbosity, data...)
2727

2828
"""
29-
MLJModelInterface.training_losses(model::M, fitresult, cache)
29+
MLJModelInterface.training_losses(model::M, report)
3030
3131
If `M` is an iterative model type which calculates training losses,
3232
implement this method to return an `AbstractVector` of the losses
@@ -37,7 +37,7 @@ The following trait overload is alse required:
3737
`supports_training_losses(::Type{<:M}) = true`
3838
3939
"""
40-
training_losses(model, fitresult, cache, report) = nothing
40+
training_losses(model, report) = nothing
4141

4242
# to support online learning in the future:
4343
# https://github.com/alan-turing-institute/MLJ.jl/issues/60 :

0 commit comments

Comments
 (0)