image.classifiers.base#

Module Contents#

Classes#

ImageClassifier

API#

class image.classifiers.base.ImageClassifier(
model_name: str,
embedding_column: str,
pred_column: str,
pred_type: str | type,
batch_size: int,
embedding_size: int,
)#

Bases: abc.ABC

Initialization

abstractmethod load_model(device: str) → collections.abc.Callable#
postprocess(series: cudf.Series) → cudf.Series#