src.superphot_plus.sfd.model.config

Module Contents

Classes

ModelConfig

Holds information about the specific training

class ModelConfig

Holds information about the specific training configuration of a model. The default values are sampled by ray tune for parameter optimization.

input_dim: int | None

output_dim: int | None

normalization_means: List[float] | None

normalization_stddevs: List[float] | None

neurons_per_layer: int | None

num_hidden_layers: int | None

goal_per_class: int | None

num_folds: int | None

num_epochs: int | None

batch_size: int | None

learning_rate: float | None

best_val_loss: float | None

device

set_non_tunable_params(input_dim, output_dim, norm_means, norm_stddevs)

Adds information about the params that are not tunable.

set_best_val_loss(best_val_loss)

Sets the best validation loss from training.

write_to_file(file: str)

Save configuration data to a YAML file.

classmethod from_file(file: str) → typing_extensions.Self

Load configuration data from a YAML file.