diff --git a/image_prediction/classifier/classifier.py b/image_prediction/classifier/classifier.py index e027292..6998bda 100644 --- a/image_prediction/classifier/classifier.py +++ b/image_prediction/classifier/classifier.py @@ -16,8 +16,7 @@ class Classifier: an EstimatorAdapter must be implemented. Args: - estimator_adapter: adapter for a given estimator backend; expected to be a classifier that returns mappings - from numeric labels to probabilities as predictions or numeric labels + estimator_adapter: adapter for a given estimator backend classes: mapping from a numerical label to a human-readable label for classes """ self.__estimator_adapter = estimator_adapter