search

unionai-oss / unionml

UnionML: the easiest way to build and deploy machine learning microservices
https://www.union.ai/unionml
Apache License 2.0
332 stars 43 forks source link

Implement model.trainer callback functions #110

Open cosmicBboy opened 2 years ago

cosmicBboy commented 2 years ago

Consider a model.callbacks.* API for defining additional behavior that’s meant to occur at some point during the training/prediction. The development should be use-case driven, e.g.

@model.trainer(pre_callback=..., post_callback=...)
def trainer(...):
    ...

Also additional callbacks for model.predict

aseemk98 commented 1 year ago

Hi @cosmicBboy, I'd be glad to take this issue!