main.py
: Main script to execute the entire pipeline. Updated parameters.modelbuilder.py
: Contains code for building and compiling models. Fixed transfer learning issues.train_utils.py
: Utilities for training models. Formatted with Black.
data_generation.ipynb
: Notebook with scripts for data generation. Added background subtraction.create_background_images.ipynb
: Scripts to generate background images.configure_dataframes.py
: Manages DataFrame configurations. Cleaned up and formatted.extract_instrument_data.py
: Scripts for specific instrument data extraction.burst_data_generation.py
: Scripts for generating burst-specific data.non_burst_data_generation.py
: Scripts for generating non-burst data.
check_data.ipynb
: Notebook to check data statistics.check_labels.ipynb
: Notebook to validate label information.check_labels.py
: Script version for checking labels.
config.yaml
: YAML configuration file. Introduced wandb support.sweep.yaml
: YAML file for wandb sweep configurations.sweep_no_folds.yaml
: Alternate sweep configuration without folds.
metric_utils.py
: Utility functions for metrics. Added batch size parameter.model_analysis.ipynb
: Notebook for model analysis using tf-explain.
requirements.txt
: List of Python dependencies.sbatch.sh
: Batch job script.sbatch_main.sh
: Main batch job script..gitignore
: Ignored files and directories.README.md
: Repository readme file. Added hello.
burst_list.xlsx
: Excel sheet containing list of bursts.data.xlsx
: Prepared data in Excel format.
original_image.png
: Sample original image.processed_image.png
: Sample processed image.
model_base_configs
: Older model configurations.