deepdrivemd.models.keras_cvae.config

pydantic settings deepdrivemd.models.keras_cvae.config.KerasCVAEModelConfig

Show JSON schema

{
   "title": "KerasCVAEModelConfig",
   "description": "Base class for specific model configs to inherit.",
   "type": "object",
   "properties": {
      "experiment_directory": {
         "title": "Experiment Directory",
         "default": "set_by_deepdrivemd",
         "env_names": "{'experiment_directory'}",
         "type": "string",
         "format": "path"
      },
      "stage_idx": {
         "title": "Stage Idx",
         "default": 0,
         "env_names": "{'stage_idx'}",
         "type": "integer"
      },
      "task_idx": {
         "title": "Task Idx",
         "default": 0,
         "env_names": "{'task_idx'}",
         "type": "integer"
      },
      "output_path": {
         "title": "Output Path",
         "default": "set_by_deepdrivemd",
         "env_names": "{'output_path'}",
         "type": "string",
         "format": "path"
      },
      "node_local_path": {
         "title": "Node Local Path",
         "default": "set_by_deepdrivemd",
         "env_names": "{'node_local_path'}",
         "type": "string",
         "format": "path"
      },
      "model_tag": {
         "title": "Model Tag",
         "default": "set_by_deepdrivemd",
         "env_names": "{'model_tag'}",
         "type": "string"
      },
      "init_weights_path": {
         "title": "Init Weights Path",
         "env_names": "{'init_weights_path'}",
         "type": "string",
         "format": "path"
      },
      "last_n_h5_files": {
         "title": "Last N H5 Files",
         "default": 10,
         "env_names": "{'last_n_h5_files'}",
         "type": "integer"
      },
      "k_random_old_h5_files": {
         "title": "K Random Old H5 Files",
         "default": 0,
         "env_names": "{'k_random_old_h5_files'}",
         "type": "integer"
      },
      "dataset_name": {
         "title": "Dataset Name",
         "default": "contact_map",
         "env_names": "{'dataset_name'}",
         "type": "string"
      },
      "initial_shape": {
         "title": "Initial Shape",
         "default": [
            28,
            28
         ],
         "env_names": "{'initial_shape'}",
         "type": "array",
         "minItems": 2,
         "maxItems": 2,
         "items": [
            {
               "type": "integer"
            },
            {
               "type": "integer"
            }
         ]
      },
      "final_shape": {
         "title": "Final Shape",
         "default": [
            28,
            28,
            1
         ],
         "env_names": "{'final_shape'}",
         "type": "array",
         "minItems": 3,
         "maxItems": 3,
         "items": [
            {
               "type": "integer"
            },
            {
               "type": "integer"
            },
            {
               "type": "integer"
            }
         ]
      },
      "initial_epochs": {
         "title": "Initial Epochs",
         "default": 10,
         "env_names": "{'initial_epochs'}",
         "type": "integer"
      },
      "epochs": {
         "title": "Epochs",
         "default": 10,
         "env_names": "{'epochs'}",
         "type": "integer"
      },
      "batch_size": {
         "title": "Batch Size",
         "default": 32,
         "env_names": "{'batch_size'}",
         "type": "integer"
      },
      "split_pct": {
         "title": "Split Pct",
         "default": 0.8,
         "env_names": "{'split_pct'}",
         "type": "number"
      },
      "shuffle": {
         "title": "Shuffle",
         "default": true,
         "env_names": "{'shuffle'}",
         "type": "boolean"
      },
      "latent_dim": {
         "title": "Latent Dim",
         "default": 10,
         "env_names": "{'latent_dim'}",
         "type": "integer"
      },
      "conv_layers": {
         "title": "Conv Layers",
         "default": 4,
         "env_names": "{'conv_layers'}",
         "type": "integer"
      },
      "conv_filters": {
         "title": "Conv Filters",
         "default": [
            64,
            64,
            64,
            64
         ],
         "env_names": "{'conv_filters'}",
         "type": "array",
         "items": {
            "type": "integer"
         }
      },
      "conv_filter_shapes": {
         "title": "Conv Filter Shapes",
         "default": [
            [
               3,
               3
            ],
            [
               3,
               3
            ],
            [
               3,
               3
            ],
            [
               3,
               3
            ]
         ],
         "env_names": "{'conv_filter_shapes'}",
         "type": "array",
         "items": {
            "type": "array",
            "minItems": 2,
            "maxItems": 2,
            "items": [
               {
                  "type": "integer"
               },
               {
                  "type": "integer"
               }
            ]
         }
      },
      "conv_strides": {
         "title": "Conv Strides",
         "default": [
            [
               1,
               1
            ],
            [
               2,
               2
            ],
            [
               1,
               1
            ],
            [
               1,
               1
            ]
         ],
         "env_names": "{'conv_strides'}",
         "type": "array",
         "items": {
            "type": "array",
            "minItems": 2,
            "maxItems": 2,
            "items": [
               {
                  "type": "integer"
               },
               {
                  "type": "integer"
               }
            ]
         }
      },
      "dense_layers": {
         "title": "Dense Layers",
         "default": 1,
         "env_names": "{'dense_layers'}",
         "type": "integer"
      },
      "dense_neurons": {
         "title": "Dense Neurons",
         "default": [
            128
         ],
         "env_names": "{'dense_neurons'}",
         "type": "array",
         "items": {
            "type": "integer"
         }
      },
      "dense_dropouts": {
         "title": "Dense Dropouts",
         "default": [
            0.25
         ],
         "env_names": "{'dense_dropouts'}",
         "type": "array",
         "items": {
            "type": "number"
         }
      },
      "use_model_checkpoint": {
         "title": "Use Model Checkpoint",
         "default": false,
         "env_names": "{'use_model_checkpoint'}",
         "type": "boolean"
      }
   }
}

Config

• extra: str = allow

Fields

• batch_size (int)

• conv_filter_shapes (List[Tuple[int, int]])

• conv_filters (List[int])

• conv_layers (int)

• conv_strides (List[Tuple[int, int]])

• dataset_name (str)

• dense_dropouts (List[float])

• dense_layers (int)

• dense_neurons (List[int])

• epochs (int)

• final_shape (Tuple[int, int, int])

• initial_epochs (int)

• initial_shape (Tuple[int, int])

• k_random_old_h5_files (int)

• last_n_h5_files (int)

• latent_dim (int)

• shuffle (bool)

• split_pct (float)

field batch_size: int = 32

field conv_filter_shapes: List[Tuple[int, int]] = [(3, 3), (3, 3), (3, 3), (3, 3)]

field conv_filters: List[int] = [64, 64, 64, 64]

field conv_layers: int = 4

field conv_strides: List[Tuple[int, int]] = [(1, 1), (2, 2), (1, 1), (1, 1)]

field dataset_name: str = 'contact_map'

field dense_dropouts: List[float] = [0.25]

field dense_layers: int = 1

field dense_neurons: List[int] = [128]

field epochs: int = 10

field final_shape: Tuple[int, int, int] = (28, 28, 1)

field initial_epochs: int = 10

field initial_shape: Tuple[int, int] = (28, 28)

field k_random_old_h5_files: int = 0

field last_n_h5_files: int = 10

field latent_dim: int = 10

field shuffle: bool = True

field split_pct: float = 0.8