stouputils.data_science.models namespace

Subpackages

• stouputils.data_science.models.keras namespace
  • Submodules
    • stouputils.data_science.models.keras.all module
    • stouputils.data_science.models.keras.convnext module
      • ConvNeXtTiny
      • ConvNeXtSmall
      • ConvNeXtBase
      • ConvNeXtLarge
      • ConvNeXtXLarge
      • model
    • stouputils.data_science.models.keras.densenet module
      • DenseNet121
      • DenseNet169
      • DenseNet201
      • model
    • stouputils.data_science.models.keras.efficientnet module
      • EfficientNetV2M
      • EfficientNetV2L
      • EfficientNetV2B0
      • EfficientNetV2S
      • EfficientNetB0
      • model
    • stouputils.data_science.models.keras.mobilenet module
      • MobileNet
      • MobileNetV2
      • MobileNetV3Small
      • MobileNetV3Large
      • model
    • stouputils.data_science.models.keras.resnet module
      • ResNet50V2
      • ResNet101V2
      • ResNet152V2
      • model
    • stouputils.data_science.models.keras.squeezenet module
      • fire_module()
      • SqueezeNet_keras()
      • SqueezeNet
      • model
    • stouputils.data_science.models.keras.vgg module
      • VGG19
      • VGG16
      • model
    • stouputils.data_science.models.keras.xception module
      • Xception
      • model
• stouputils.data_science.models.keras_utils namespace
  • Subpackages
    • stouputils.data_science.models.keras_utils.callbacks package
      • Submodules
    • stouputils.data_science.models.keras_utils.losses package
      • Submodules
  • Submodules
    • stouputils.data_science.models.keras_utils.visualizations module
      • make_gradcam_heatmap()
      • make_saliency_map()
      • find_last_conv_layer()
      • create_visualization_overlay()
      • all_visualizations_for_image()

Submodules

• stouputils.data_science.models.abstract_model module
  • AbstractModel
    • AbstractModel.routine_full()
    • AbstractModel.class_load()
    • AbstractModel.class_train()
    • AbstractModel.class_predict()
    • AbstractModel.class_evaluate()
    • AbstractModel._fit()
    • AbstractModel._get_callbacks()
    • AbstractModel._get_metrics()
    • AbstractModel._get_optimizer()
    • AbstractModel._get_loss()
    • AbstractModel._get_base_model()
    • AbstractModel._get_architectures()
    • AbstractModel._find_best_learning_rate()
    • AbstractModel._train_fold()
    • AbstractModel._log_final_model()
    • AbstractModel._find_best_learning_rate_subprocess()
    • AbstractModel._find_best_unfreeze_percentage_subprocess()
    • AbstractModel._train_subprocess()
• stouputils.data_science.models.all module
  • ModelClassMap
  • ALL_MODELS
• stouputils.data_science.models.base_keras module
  • BaseKeras
    • BaseKeras.class_load()
    • BaseKeras._fit()
    • BaseKeras._get_architectures()
    • BaseKeras._get_callbacks()
    • BaseKeras._get_metrics()
    • BaseKeras._get_optimizer()
    • BaseKeras._get_loss()
    • BaseKeras._find_best_learning_rate_subprocess()
    • BaseKeras._find_best_unfreeze_percentage_subprocess()
    • BaseKeras._train_subprocess()
    • BaseKeras.class_predict()
    • BaseKeras._log_final_model()
    • BaseKeras.class_evaluate()
• stouputils.data_science.models.model_interface module
  • ModelInterface
    • ModelInterface.final_model
    • ModelInterface.model_name
    • ModelInterface.kfold
    • ModelInterface.transfer_learning
    • ModelInterface.is_trained
    • ModelInterface.num_classes
    • ModelInterface.override_params
    • ModelInterface.run_name
    • ModelInterface.history
    • ModelInterface.evaluation_results
    • ModelInterface.additional_training_data
    • ModelInterface.batch_size
    • ModelInterface.epochs
    • ModelInterface.class_weight
    • ModelInterface.unfreeze_percentage
    • ModelInterface.fine_tune_last_layers
    • ModelInterface.beta_1
    • ModelInterface.beta_2
    • ModelInterface.early_stop_patience
    • ModelInterface.model_checkpoint_delay
    • ModelInterface.learning_rate
    • ModelInterface.reduce_lr_patience
    • ModelInterface.min_delta
    • ModelInterface.min_lr
    • ModelInterface.factor
    • ModelInterface.warmup_epochs
    • ModelInterface.initial_warmup_lr
    • ModelInterface.lr_finder_min_lr
    • ModelInterface.lr_finder_max_lr
    • ModelInterface.lr_finder_epochs
    • ModelInterface.lr_finder_update_per_epoch
    • ModelInterface.lr_finder_update_interval
    • ModelInterface.unfreeze_finder_epochs
    • ModelInterface.unfreeze_finder_update_per_epoch
    • ModelInterface.unfreeze_finder_update_interval
    • ModelInterface.total_layers
    • ModelInterface.class_load()
    • ModelInterface.train()
    • ModelInterface.predict()
    • ModelInterface.evaluate()
    • ModelInterface.routine_full()
    • ModelInterface._routine()
    • ModelInterface._get_transfer_learning_weights()
    • ModelInterface._get_total_layers()
    • ModelInterface._set_parameters()
    • ModelInterface._set_class_weight()
    • ModelInterface._log_parameters()
    • ModelInterface._get_fold_split()
    • ModelInterface._train_final_model()
    • ModelInterface._train_each_fold()
    • ModelInterface.class_train()
    • ModelInterface._log_metrics()
    • ModelInterface.class_evaluate()
    • ModelInterface._find_best_learning_rate()
    • ModelInterface._find_best_unfreeze_percentage()
    • ModelInterface._train_fold()
• stouputils.data_science.models.sandbox module
  • Sandbox
    • Sandbox.custom_architecture()
    • Sandbox.class_routine()
  • model