14 lines
255 B
Python
14 lines
255 B
Python
import abc
|
|
|
|
|
|
class Model(abc.ABC):
|
|
def __init__(self, estimator):
|
|
self.__estimator = estimator
|
|
|
|
@property
|
|
def estimator(self):
|
|
return self.__estimator
|
|
|
|
def predict(self, batch):
|
|
return self.estimator.predict(batch)
|