search

ShannonAI / service-streamer

Boosting your Web Services of Deep Learning Applications.
Apache License 2.0
1.22k stars 187 forks source link

Passing additional arguments to predict function #56

Closed radao closed 4 years ago

radao commented 4 years ago

Just had a quick question, is passing additional arguments to the predict function on the roadmap? What I'm looking for is a way to pass an additional argument, such as a threshold, to the model_predict call of the _BaseStreamWorker class.

radao commented 4 years ago

This can be very easily dealt with by passing dictionaries/tuples to your predict function and unpacking them there so closing the issue.