Callback at end of `learn.predict`: ''on_predict_end"?


I have several callbacks that collect metrics when I do learn.predict().
I would like to collect them at the end of that process without adding extra code in my notebook/production code. Could I add a PR that adds a on_predict_end() callback option? It would make sense to me given that on_train_end() exists as well.
Or is there another way to do this without using such a callback?

(My final goal is to create a small monitoring app that sends these metrics to a dashboard - on_predict_end() could then be used to collect all metrics and send them to a database)

There will be the equivalent in v2. We are not making any major change to v1 at this stage,

1 Like