Source code for src.superphot_plus.file_paths

"""A collection of paths corresponding to input and output files and directories."""

[docs]DATA_DIR = "data"

[docs]FITS_DIR = "/storage/group/vav5084/default/superphot+/dynesty_fits_unclassified_5_9_2023/"
[docs]DATA_DIRS = [
    "/storage/group/vav5084/default/superphot+/data_reformatted_05_09_2023",
]
[docs]INPUT_CSVS = [f"{DATA_DIR}/training_set.csv"]  # ["../data/training_set_combined_05_09_2023.csv"]

# Directory to store models
[docs]MODELS_DIR = f"{DATA_DIR}/models"

# Directories to store metrics
[docs]METRICS_DIR = f"{DATA_DIR}/figs/metrics"
[docs]FIT_PLOTS_FOLDER = f"{DATA_DIR}/figs/fits_good"
[docs]CM_FOLDER = f"{DATA_DIR}/figs/hp_cm"
[docs]WRONGLY_CLASSIFIED_FOLDER = f"{DATA_DIR}/figs/wrongly_classified"

# Classification output files
[docs]CLASSIFY_LOG_FILE = f"{DATA_DIR}/classification_log.txt"
[docs]PROBS_FILE = f"{DATA_DIR}/probs_new.csv"
[docs]PROBS_FILE2 = f"{DATA_DIR}/probs_phased.csv"

# Pretrained model
# glob.glob("models_saved/*ZTF23aagkgnz.pt")[0]
[docs]TRAINED_MODEL_FN = "../../tests/data/superphot-model-ZTF23aagkgnz.pt"
[docs]TRAINED_CONFIG_FN = "../../tests/data/superphot-config-test.yaml"