stouputils 1.14.0__py3-none-any.whl → 1.14.2__py3-none-any.whl

stouputils/data_science/models/abstract_model.py

@@ -1,149 +1,149 @@
-""" Abstract base class for all model implementations.
-Defines the interface that all concrete model classes must implement.
-
-Provides abstract methods for core model operations including:
-
-- Class routine management
-- Model loading
-- Training procedures
-- Prediction functionality
-- Evaluation metrics
-
-Classes inheriting from AbstractModel must implement all methods.
-"""
-
-# Imports
-from __future__ import annotations
-
-import multiprocessing.queues
-from collections.abc import Iterable
-from tempfile import TemporaryDirectory
-from typing import Any
-
-from ...decorators import abstract, LogLevels
-
-from ..dataset import Dataset
-
-
-# Base class
-class AbstractModel:
-	""" Abstract class for all models to copy and implement the methods. """
-	# Class methods
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def __init__(
-		self, num_classes: int, kfold: int = 0, transfer_learning: str = "imagenet", **override_params: Any
-	) -> None:
-		pass
-
-
-	## Public abstract methods
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def routine_full(self, dataset: Dataset, verbose: int = 0) -> AbstractModel:
-		return self
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def class_load(self) -> None:
-		pass
-
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def class_train(self, dataset: Dataset, verbose: int = 0) -> bool:
-		return False
-
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def class_predict(self, X_test: Iterable[Any]) -> Iterable[Any]:
-		return []
-
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def class_evaluate(
-		self,
-		dataset: Dataset,
-		metrics_names: tuple[str, ...] = (),
-		save_model: bool = False,
-		verbose: int = 0
-	) -> bool:
-		return False
-
-
-	## Protected abstract methods
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _fit(
-		self,
-		model: Any,
-		x: Any,
-		y: Any | None = None,
-		validation_data: tuple[Any, Any] | None = None,
-		shuffle: bool = True,
-		batch_size: int | None = None,
-		epochs: int = 1,
-		callbacks: list[Any] | None = None,
-		class_weight: dict[int, float] | None = None,
-		verbose: int = 0,
-		*args: Any,
-		**kwargs: Any
-	) -> Any:
-		pass
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _get_callbacks(self) -> list[Any]:
-		return []
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _get_metrics(self) -> list[Any]:
-		return []
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _get_optimizer(self, learning_rate: float = 0.0) -> Any:
-		pass
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _get_loss(self) -> Any:
-		pass
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _get_base_model(self) -> Any:
-		pass
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _get_architectures(
-		self, optimizer: Any = None, loss: Any = None, metrics: list[Any] | None = None
-	) -> tuple[Any, Any]:
-		return (None, None)
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _find_best_learning_rate(self, dataset: Dataset, verbose: int = 0) -> float:
-		return 0.0
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _train_fold(self, dataset: Dataset, fold_number: int = 0, mlflow_prefix: str = "history", verbose: int = 0) -> Any:
-		pass
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _log_final_model(self) -> None:
-		pass
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _find_best_learning_rate_subprocess(
-		self, dataset: Dataset, queue: multiprocessing.queues.Queue[Any] | None = None, verbose: int = 0
-	) -> dict[str, Any] | None:
-		pass
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _find_best_unfreeze_percentage_subprocess(
-		self, dataset: Dataset, queue: multiprocessing.queues.Queue[Any] | None = None, verbose: int = 0
-	) -> dict[str, Any] | None:
-		pass
-
-	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
-	def _train_subprocess(
-		self,
-		dataset: Dataset,
-		checkpoint_path: str,
-		temp_dir: TemporaryDirectory[str] | None = None,
-		queue: multiprocessing.queues.Queue[Any] | None = None,
-		verbose: int = 0
-	) -> dict[str, Any] | None:
-		pass
-
+""" Abstract base class for all model implementations.
+Defines the interface that all concrete model classes must implement.
+
+Provides abstract methods for core model operations including:
+
+- Class routine management
+- Model loading
+- Training procedures
+- Prediction functionality
+- Evaluation metrics
+
+Classes inheriting from AbstractModel must implement all methods.
+"""
+
+# Imports
+from __future__ import annotations
+
+import multiprocessing.queues
+from collections.abc import Iterable
+from tempfile import TemporaryDirectory
+from typing import Any
+
+from ...decorators import abstract, LogLevels
+
+from ..dataset import Dataset
+
+
+# Base class
+class AbstractModel:
+	""" Abstract class for all models to copy and implement the methods. """
+	# Class methods
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def __init__(
+		self, num_classes: int, kfold: int = 0, transfer_learning: str = "imagenet", **override_params: Any
+	) -> None:
+		pass
+
+
+	## Public abstract methods
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def routine_full(self, dataset: Dataset, verbose: int = 0) -> AbstractModel:
+		return self
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def class_load(self) -> None:
+		pass
+
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def class_train(self, dataset: Dataset, verbose: int = 0) -> bool:
+		return False
+
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def class_predict(self, X_test: Iterable[Any]) -> Iterable[Any]:
+		return []
+
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def class_evaluate(
+		self,
+		dataset: Dataset,
+		metrics_names: tuple[str, ...] = (),
+		save_model: bool = False,
+		verbose: int = 0
+	) -> bool:
+		return False
+
+
+	## Protected abstract methods
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _fit(
+		self,
+		model: Any,
+		x: Any,
+		y: Any | None = None,
+		validation_data: tuple[Any, Any] | None = None,
+		shuffle: bool = True,
+		batch_size: int | None = None,
+		epochs: int = 1,
+		callbacks: list[Any] | None = None,
+		class_weight: dict[int, float] | None = None,
+		verbose: int = 0,
+		*args: Any,
+		**kwargs: Any
+	) -> Any:
+		pass
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _get_callbacks(self) -> list[Any]:
+		return []
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _get_metrics(self) -> list[Any]:
+		return []
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _get_optimizer(self, learning_rate: float = 0.0) -> Any:
+		pass
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _get_loss(self) -> Any:
+		pass
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _get_base_model(self) -> Any:
+		pass
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _get_architectures(
+		self, optimizer: Any = None, loss: Any = None, metrics: list[Any] | None = None
+	) -> tuple[Any, Any]:
+		return (None, None)
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _find_best_learning_rate(self, dataset: Dataset, verbose: int = 0) -> float:
+		return 0.0
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _train_fold(self, dataset: Dataset, fold_number: int = 0, mlflow_prefix: str = "history", verbose: int = 0) -> Any:
+		pass
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _log_final_model(self) -> None:
+		pass
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _find_best_learning_rate_subprocess(
+		self, dataset: Dataset, queue: multiprocessing.queues.Queue[Any] | None = None, verbose: int = 0
+	) -> dict[str, Any] | None:
+		pass
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _find_best_unfreeze_percentage_subprocess(
+		self, dataset: Dataset, queue: multiprocessing.queues.Queue[Any] | None = None, verbose: int = 0
+	) -> dict[str, Any] | None:
+		pass
+
+	@abstract(error_log=LogLevels.ERROR_TRACEBACK)
+	def _train_subprocess(
+		self,
+		dataset: Dataset,
+		checkpoint_path: str,
+		temp_dir: TemporaryDirectory[str] | None = None,
+		queue: multiprocessing.queues.Queue[Any] | None = None,
+		verbose: int = 0
+	) -> dict[str, Any] | None:
+		pass
+

stouputils/data_science/models/all.py

@@ -1,85 +1,85 @@
-
-# Imports
-import itertools
-
-from .keras.all import (
-    VGG16,
-    VGG19,
-    ConvNeXtBase,
-    ConvNeXtLarge,
-    ConvNeXtSmall,
-    ConvNeXtTiny,
-    ConvNeXtXLarge,
-    DenseNet121,
-    DenseNet169,
-    DenseNet201,
-    EfficientNetB0,
-    EfficientNetV2B0,
-    EfficientNetV2L,
-    EfficientNetV2M,
-    EfficientNetV2S,
-    MobileNet,
-    MobileNetV2,
-    MobileNetV3Large,
-    MobileNetV3Small,
-    ResNet50V2,
-    ResNet101V2,
-    ResNet152V2,
-    SqueezeNet,
-    Xception,
-)
-from .model_interface import ModelInterface
-
-# Other models
-from .sandbox import Sandbox
-
-
-# Create a custom dictionary class to allow for documentation
-class ModelClassMap(dict[type[ModelInterface], tuple[str, ...]]):
-    pass
-
-# Routine map
-CLASS_MAP: ModelClassMap = ModelClassMap({
-    SqueezeNet:         ("squeezenet", "squeezenets", "all", "often"),
-
-    DenseNet121:        ("densenet121", "densenets", "all", "often", "good"),
-    DenseNet169:        ("densenet169", "densenets", "all", "often", "good"),
-    DenseNet201:        ("densenet201", "densenets", "all", "often", "good"),
-
-    EfficientNetB0:     ("efficientnetb0", "efficientnets", "all"),
-    EfficientNetV2B0:   ("efficientnetv2b0", "efficientnets", "all"),
-    EfficientNetV2S:    ("efficientnetv2s", "efficientnets", "all", "often"),
-    EfficientNetV2M:    ("efficientnetv2m", "efficientnets", "all", "often"),
-    EfficientNetV2L:    ("efficientnetv2l", "efficientnets", "all", "often"),
-
-    ConvNeXtTiny:       ("convnexttiny", "convnexts", "all", "often", "good"),
-    ConvNeXtSmall:      ("convnextsmall", "convnexts", "all", "often"),
-    ConvNeXtBase:       ("convnextbase", "convnexts", "all", "often", "good"),
-    ConvNeXtLarge:      ("convnextlarge", "convnexts", "all", "often"),
-    ConvNeXtXLarge:     ("convnextxlarge", "convnexts", "all", "often", "good"),
-
-    VGG16:              ("vgg16", "vggs", "all"),
-    VGG19:              ("vgg19", "vggs", "all"),
-
-    MobileNet:          ("mobilenet", "mobilenets", "all"),
-    MobileNetV2:        ("mobilenetv2", "mobilenets", "all", "often"),
-    MobileNetV3Small:   ("mobilenetv3small", "mobilenets", "all", "often"),
-    MobileNetV3Large:   ("mobilenetv3large", "mobilenets", "all", "often", "good"),
-
-    ResNet50V2:         ("resnet50v2", "resnetsv2", "resnets", "all", "often"),
-    ResNet101V2:        ("resnet101v2", "resnetsv2", "resnets", "all", "often"),
-    ResNet152V2:        ("resnet152v2", "resnetsv2", "resnets", "all", "often"),
-
-    Xception:           ("xception", "xceptions", "all", "often"),
-    Sandbox:            ("sandbox",),
-})
-
-# All models names and aliases
-ALL_MODELS: list[str] = sorted(set(itertools.chain.from_iterable(v for v in CLASS_MAP.values())))
-""" All models names and aliases found in the `CLASS_MAP` dictionary. """
-
-# Additional docstring
-new_docstring: str = "\n\n" + "\n".join(f"- {k.__name__}: {v}" for k, v in CLASS_MAP.items())
-ModelClassMap.__doc__ = "Dictionary mapping class to their names and aliases. " + new_docstring
-CLASS_MAP.__doc__ = ModelClassMap.__doc__
-
+
+# Imports
+import itertools
+
+from .keras.all import (
+    VGG16,
+    VGG19,
+    ConvNeXtBase,
+    ConvNeXtLarge,
+    ConvNeXtSmall,
+    ConvNeXtTiny,
+    ConvNeXtXLarge,
+    DenseNet121,
+    DenseNet169,
+    DenseNet201,
+    EfficientNetB0,
+    EfficientNetV2B0,
+    EfficientNetV2L,
+    EfficientNetV2M,
+    EfficientNetV2S,
+    MobileNet,
+    MobileNetV2,
+    MobileNetV3Large,
+    MobileNetV3Small,
+    ResNet50V2,
+    ResNet101V2,
+    ResNet152V2,
+    SqueezeNet,
+    Xception,
+)
+from .model_interface import ModelInterface
+
+# Other models
+from .sandbox import Sandbox
+
+
+# Create a custom dictionary class to allow for documentation
+class ModelClassMap(dict[type[ModelInterface], tuple[str, ...]]):
+    pass
+
+# Routine map
+CLASS_MAP: ModelClassMap = ModelClassMap({
+    SqueezeNet:         ("squeezenet", "squeezenets", "all", "often"),
+
+    DenseNet121:        ("densenet121", "densenets", "all", "often", "good"),
+    DenseNet169:        ("densenet169", "densenets", "all", "often", "good"),
+    DenseNet201:        ("densenet201", "densenets", "all", "often", "good"),
+
+    EfficientNetB0:     ("efficientnetb0", "efficientnets", "all"),
+    EfficientNetV2B0:   ("efficientnetv2b0", "efficientnets", "all"),
+    EfficientNetV2S:    ("efficientnetv2s", "efficientnets", "all", "often"),
+    EfficientNetV2M:    ("efficientnetv2m", "efficientnets", "all", "often"),
+    EfficientNetV2L:    ("efficientnetv2l", "efficientnets", "all", "often"),
+
+    ConvNeXtTiny:       ("convnexttiny", "convnexts", "all", "often", "good"),
+    ConvNeXtSmall:      ("convnextsmall", "convnexts", "all", "often"),
+    ConvNeXtBase:       ("convnextbase", "convnexts", "all", "often", "good"),
+    ConvNeXtLarge:      ("convnextlarge", "convnexts", "all", "often"),
+    ConvNeXtXLarge:     ("convnextxlarge", "convnexts", "all", "often", "good"),
+
+    VGG16:              ("vgg16", "vggs", "all"),
+    VGG19:              ("vgg19", "vggs", "all"),
+
+    MobileNet:          ("mobilenet", "mobilenets", "all"),
+    MobileNetV2:        ("mobilenetv2", "mobilenets", "all", "often"),
+    MobileNetV3Small:   ("mobilenetv3small", "mobilenets", "all", "often"),
+    MobileNetV3Large:   ("mobilenetv3large", "mobilenets", "all", "often", "good"),
+
+    ResNet50V2:         ("resnet50v2", "resnetsv2", "resnets", "all", "often"),
+    ResNet101V2:        ("resnet101v2", "resnetsv2", "resnets", "all", "often"),
+    ResNet152V2:        ("resnet152v2", "resnetsv2", "resnets", "all", "often"),
+
+    Xception:           ("xception", "xceptions", "all", "often"),
+    Sandbox:            ("sandbox",),
+})
+
+# All models names and aliases
+ALL_MODELS: list[str] = sorted(set(itertools.chain.from_iterable(v for v in CLASS_MAP.values())))
+""" All models names and aliases found in the `CLASS_MAP` dictionary. """
+
+# Additional docstring
+new_docstring: str = "\n\n" + "\n".join(f"- {k.__name__}: {v}" for k, v in CLASS_MAP.items())
+ModelClassMap.__doc__ = "Dictionary mapping class to their names and aliases. " + new_docstring
+CLASS_MAP.__doc__ = ModelClassMap.__doc__
+

stouputils/data_science/models/keras/all.py

@@ -1,38 +1,38 @@
-
-# Imports
-from .convnext import ConvNeXtBase, ConvNeXtLarge, ConvNeXtSmall, ConvNeXtTiny, ConvNeXtXLarge
-from .densenet import DenseNet121, DenseNet169, DenseNet201
-from .efficientnet import EfficientNetB0, EfficientNetV2B0, EfficientNetV2L, EfficientNetV2M, EfficientNetV2S
-from .mobilenet import MobileNet, MobileNetV2, MobileNetV3Large, MobileNetV3Small
-from .resnet import ResNet50V2, ResNet101V2, ResNet152V2
-from .squeezenet import SqueezeNet
-from .vgg import VGG16, VGG19
-from .xception import Xception
-
-__all__ = [
-    "VGG16",
-    "VGG19",
-    "ConvNeXtBase",
-    "ConvNeXtLarge",
-    "ConvNeXtSmall",
-    "ConvNeXtTiny",
-    "ConvNeXtXLarge",
-    "DenseNet121",
-    "DenseNet169",
-    "DenseNet201",
-    "EfficientNetB0",
-    "EfficientNetV2B0",
-    "EfficientNetV2L",
-    "EfficientNetV2M",
-    "EfficientNetV2S",
-    "MobileNet",
-    "MobileNetV2",
-    "MobileNetV3Large",
-    "MobileNetV3Small",
-    "ResNet50V2",
-    "ResNet101V2",
-    "ResNet152V2",
-    "SqueezeNet",
-    "Xception",
-]
-
+
+# Imports
+from .convnext import ConvNeXtBase, ConvNeXtLarge, ConvNeXtSmall, ConvNeXtTiny, ConvNeXtXLarge
+from .densenet import DenseNet121, DenseNet169, DenseNet201
+from .efficientnet import EfficientNetB0, EfficientNetV2B0, EfficientNetV2L, EfficientNetV2M, EfficientNetV2S
+from .mobilenet import MobileNet, MobileNetV2, MobileNetV3Large, MobileNetV3Small
+from .resnet import ResNet50V2, ResNet101V2, ResNet152V2
+from .squeezenet import SqueezeNet
+from .vgg import VGG16, VGG19
+from .xception import Xception
+
+__all__ = [
+    "VGG16",
+    "VGG19",
+    "ConvNeXtBase",
+    "ConvNeXtLarge",
+    "ConvNeXtSmall",
+    "ConvNeXtTiny",
+    "ConvNeXtXLarge",
+    "DenseNet121",
+    "DenseNet169",
+    "DenseNet201",
+    "EfficientNetB0",
+    "EfficientNetV2B0",
+    "EfficientNetV2L",
+    "EfficientNetV2M",
+    "EfficientNetV2S",
+    "MobileNet",
+    "MobileNetV2",
+    "MobileNetV3Large",
+    "MobileNetV3Small",
+    "ResNet50V2",
+    "ResNet101V2",
+    "ResNet152V2",
+    "SqueezeNet",
+    "Xception",
+]
+

stouputils/data_science/models/keras/convnext.py

@@ -1,62 +1,62 @@
-""" ConvNeXt models implementation.
-
-This module provides wrapper classes for the ConvNeXt family of models from the Keras applications.
-ConvNeXt models are a family of pure convolutional networks that match or outperform
-Vision Transformers (ViTs) while maintaining the simplicity and efficiency of CNNs.
-
-Available models:
-
-- ConvNeXtTiny: Smallest variant with fewer parameters for resource-constrained environments
-- ConvNeXtSmall: Compact model balancing performance and size
-- ConvNeXtBase: Standard model with good performance for general use cases
-- ConvNeXtLarge: Larger model with higher capacity for complex tasks
-- ConvNeXtXLarge: Largest variant with maximum capacity for demanding applications
-
-All models support transfer learning from ImageNet pre-trained weights.
-"""
-# pyright: reportUnknownVariableType=false
-# pyright: reportMissingTypeStubs=false
-
-# Imports
-from __future__ import annotations
-
-from keras.models import Model
-from keras.src.applications.convnext import ConvNeXtBase as ConvNeXtBase_keras
-from keras.src.applications.convnext import ConvNeXtLarge as ConvNeXtLarge_keras
-from keras.src.applications.convnext import ConvNeXtSmall as ConvNeXtSmall_keras
-from keras.src.applications.convnext import ConvNeXtTiny as ConvNeXtTiny_keras
-from keras.src.applications.convnext import ConvNeXtXLarge as ConvNeXtXLarge_keras
-
-from ....decorators import simple_cache
-from ..base_keras import BaseKeras
-from ..model_interface import CLASS_ROUTINE_DOCSTRING, MODEL_DOCSTRING
-
-
-# Classes
-class ConvNeXtTiny(BaseKeras):
-	def _get_base_model(self) -> Model:
-		return ConvNeXtTiny_keras(include_top=False, classes=self.num_classes)
-
-class ConvNeXtSmall(BaseKeras):
-	def _get_base_model(self) -> Model:
-		return ConvNeXtSmall_keras(include_top=False, classes=self.num_classes)
-
-class ConvNeXtBase(BaseKeras):
-	def _get_base_model(self) -> Model:
-		return ConvNeXtBase_keras(include_top=False, classes=self.num_classes)
-
-class ConvNeXtLarge(BaseKeras):
-	def _get_base_model(self) -> Model:
-		return ConvNeXtLarge_keras(include_top=False, classes=self.num_classes)
-
-class ConvNeXtXLarge(BaseKeras):
-	def _get_base_model(self) -> Model:
-		return ConvNeXtXLarge_keras(include_top=False, classes=self.num_classes)
-
-
-# Docstrings
-for model in [ConvNeXtTiny, ConvNeXtSmall, ConvNeXtBase, ConvNeXtLarge, ConvNeXtXLarge]:
-	model.__doc__ = MODEL_DOCSTRING.format(model=model.__name__)
-	model.class_routine = simple_cache(model.class_routine)
-	model.class_routine.__doc__ = CLASS_ROUTINE_DOCSTRING.format(model=model.__name__)
-
+""" ConvNeXt models implementation.
+
+This module provides wrapper classes for the ConvNeXt family of models from the Keras applications.
+ConvNeXt models are a family of pure convolutional networks that match or outperform
+Vision Transformers (ViTs) while maintaining the simplicity and efficiency of CNNs.
+
+Available models:
+
+- ConvNeXtTiny: Smallest variant with fewer parameters for resource-constrained environments
+- ConvNeXtSmall: Compact model balancing performance and size
+- ConvNeXtBase: Standard model with good performance for general use cases
+- ConvNeXtLarge: Larger model with higher capacity for complex tasks
+- ConvNeXtXLarge: Largest variant with maximum capacity for demanding applications
+
+All models support transfer learning from ImageNet pre-trained weights.
+"""
+# pyright: reportUnknownVariableType=false
+# pyright: reportMissingTypeStubs=false
+
+# Imports
+from __future__ import annotations
+
+from keras.models import Model
+from keras.src.applications.convnext import ConvNeXtBase as ConvNeXtBase_keras
+from keras.src.applications.convnext import ConvNeXtLarge as ConvNeXtLarge_keras
+from keras.src.applications.convnext import ConvNeXtSmall as ConvNeXtSmall_keras
+from keras.src.applications.convnext import ConvNeXtTiny as ConvNeXtTiny_keras
+from keras.src.applications.convnext import ConvNeXtXLarge as ConvNeXtXLarge_keras
+
+from ....decorators import simple_cache
+from ..base_keras import BaseKeras
+from ..model_interface import CLASS_ROUTINE_DOCSTRING, MODEL_DOCSTRING
+
+
+# Classes
+class ConvNeXtTiny(BaseKeras):
+	def _get_base_model(self) -> Model:
+		return ConvNeXtTiny_keras(include_top=False, classes=self.num_classes)
+
+class ConvNeXtSmall(BaseKeras):
+	def _get_base_model(self) -> Model:
+		return ConvNeXtSmall_keras(include_top=False, classes=self.num_classes)
+
+class ConvNeXtBase(BaseKeras):
+	def _get_base_model(self) -> Model:
+		return ConvNeXtBase_keras(include_top=False, classes=self.num_classes)
+
+class ConvNeXtLarge(BaseKeras):
+	def _get_base_model(self) -> Model:
+		return ConvNeXtLarge_keras(include_top=False, classes=self.num_classes)
+
+class ConvNeXtXLarge(BaseKeras):
+	def _get_base_model(self) -> Model:
+		return ConvNeXtXLarge_keras(include_top=False, classes=self.num_classes)
+
+
+# Docstrings
+for model in [ConvNeXtTiny, ConvNeXtSmall, ConvNeXtBase, ConvNeXtLarge, ConvNeXtXLarge]:
+	model.__doc__ = MODEL_DOCSTRING.format(model=model.__name__)
+	model.class_routine = simple_cache(model.class_routine)
+	model.class_routine.__doc__ = CLASS_ROUTINE_DOCSTRING.format(model=model.__name__)
+