InseeFrLab / InseeFrLab/torchTextClassifiers

Provide a generic mlflow pyfunc wrapper

Aperta
#56 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub
enhancement
Lingua principale
Python
Stelle
23
Fork
7
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

When using [mlflow](https://mlflow.org/) to train and register a text classifier model using the package, we often have to write our own pyfunc wrapper, which could be tedious.

The idea would be to provide a generic wrapper (with some hypothesis on the stored artifact) usable by all package users.

As a bonus, it could be really great if this wrapper was self sufficient (no need to install the package for inference, only torch and numpy)

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.