adding callback to init

rodrigo-arenas · Oct 4, 2024 · c5926d9 · c5926d9
1 parent f6b0386
commit c5926d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sklearn_genetic/callbacks/__init__.py b/sklearn_genetic/callbacks/__init__.py
@@ -5,6 +5,7 @@
     TimerStopping,
 )
 from .loggers import ProgressBar, LogbookSaver, TensorBoard
+from .model_checkpoint import ModelCheckpoint
 
 __all__ = [
     "ProgressBar",
@@ -14,4 +15,5 @@
     "TimerStopping",
     "LogbookSaver",
     "TensorBoard",
+    "ModelCheckpoint"
 ]