15 lines
437 B
Python
15 lines
437 B
Python
from funcy import rcompose
|
|
|
|
|
|
class PredictionModelHandle:
|
|
"""Simplifies usage of ModelHandle instances for prediction purposes."""
|
|
|
|
def __init__(self, model_handle):
|
|
self.__predict = rcompose(model_handle.prep_images, model_handle.model.predict)
|
|
|
|
def predict(self, *args, **kwargs):
|
|
return self.__predict(*args, **kwargs)
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
return self.predict(*args, **kwargs)
|