12 lines
264 B
Python
12 lines
264 B
Python
import numpy as np
|
|
|
|
from image_prediction.estimator.estimator import Estimator
|
|
|
|
|
|
class KerasEstimator(Estimator):
|
|
def __init__(self, estimator):
|
|
super().__init__(estimator)
|
|
|
|
def predict(self, batch: np.array):
|
|
self.estimator.predict(batch)
|