12 lines
298 B
Python
12 lines
298 B
Python
import numpy as np
|
|
|
|
from image_prediction.estimator.adapter.adapter import EstimatorAdapter
|
|
|
|
|
|
class KerasEstimatorAdapter(EstimatorAdapter):
|
|
def __init__(self, estimator):
|
|
super().__init__(estimator)
|
|
|
|
def predict(self, batch: np.array):
|
|
return self.estimator.predict(batch)
|