Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

renaming of service estimator to estimator

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-25 18:20:44 +01:00
parent dd18087261
commit c64bff0843
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
image_prediction/estimator/preprocessor/preprocessor.py
View File

@ -1,4 +1,4 @@
from image_prediction.estimator.adapter import EstimatorAdapter
from image_prediction.estimator.adapter.adapter import EstimatorAdapter
class EstimatorPreprocessor:

4
image_prediction/estimator/service_estimator.py
View File

@ -8,7 +8,7 @@ from image_prediction.utils import get_logger
logger = get_logger()
class ServiceEstimator:
class Estimator:
def __init__(self, estimator_adapter: EstimatorAdapter, classes: Mapping[int, str]):
self.__estimator_adapter = estimator_adapter
self.__classes = classes
@ -16,7 +16,7 @@ class ServiceEstimator:
def predict(self, batch: np.array) -> List[str]:
if batch.shape[0] == 0:
logger.warning("ServiceEstimator received empty batch.")
logger.warning("Estimator received empty batch.")
return []
return [self.__classes[numeric_label] for numeric_label in self.__estimator_adapter.predict(batch)]

4
test/unit_tests/conftest.py
View File

@ -6,7 +6,7 @@ from image_prediction.estimator.adapter.adapters.keras import KerasEstimatorAdap
from image_prediction.estimator.adapter.adapters.mock import DummyEstimator, EstimatorAdapterMock
from image_prediction.exceptions import UnknownEstimatorAdapter
from image_prediction.predictor.predictor import Predictor
from image_prediction.estimator.service_estimator import ServiceEstimator
from image_prediction.estimator.service_estimator import Estimator
@pytest.fixture
@ -16,7 +16,7 @@ def predictor(service_estimator):
@pytest.fixture
def service_estimator(estimator, classes):
service_estimator = ServiceEstimator(estimator, classes)
service_estimator = Estimator(estimator, classes)
return service_estimator