BASELINES_PATH = "/app2/suno/data/ditto_evals/baselines_2"

diversity_scores = {
    "Auk": "div_feat_eval_diversity_chirp-auk-t1_self_sim_2025_09_17-19_56_25.npz",
    "BlueJay": "div_feat_eval_diversity_chirp-bluejay-t2_self_sim_2025_09_16-18_58_57.npz",
    "Crow d87": "crow-a-r87_div_feat_eval_diversity_chirp-crow-eval_self_sim_2025_09_21-00_00_50.npz",
    "Crow d87+2": "crow-d87+2_div_feat_eval_diversity_chirp-crow-eval_self_sim_2025_09_22-15_46_55.npz",
    # "Crow d89": "crow-d89_div_feat_eval_diversity_chirp-crow-eval_self_sim_2025_09_22-13_58_22.npz",
    # "Crow d89+4": "crow-d89+4_div_feat_eval_diversity_chirp-crow-eval_self_sim_2025_09_22-14_47_09.npz",
    # "Crow d101": "crow-a-r11_div_feat_eval_diversity_chirp-crow-eval_self_sim_2025_09_20-21_25_45.npz",
    # "Crow d102": "crow-a-r19_div_feat_eval_diversity_chirp-crow-eval_self_sim_2025_09_20-22_16_38.npz",
}

drift_scores = {
    "Auk": "div_feat_eval_drift_chirp-auk-t1_2025_09_17-19_56_25.npz",
    "BlueJay": "div_feat_eval_drift_chirp-bluejay-t2_2025_09_16-18_58_57.npz",
    "Crow d87": "crow-a-r87_div_feat_eval_drift_chirp-crow-eval_2025_09_21-00_00_50.npz",
    "Crow d87+2": "crow-d87+2_div_feat_eval_drift_chirp-crow-eval_2025_09_22-15_46_55.npz",
    # "Crow d89": "crow-d89_div_feat_eval_drift_chirp-crow-eval_2025_09_22-13_58_22.npz",
    # "Crow d89+4": "crow-d89+4_div_feat_eval_drift_chirp-crow-eval_2025_09_22-14_47_09.npz",
    # "Crow d101": "crow-a-r11_div_feat_eval_drift_chirp-crow-eval_2025_09_20-21_25_45.npz",
    # "Crow d102": "crow-a-r19_div_feat_eval_drift_chirp-crow-eval_2025_09_20-22_16_38.npz",
}

INFILL_SCORES = {
    "30b": "30b_2025_07_15-19_14_15_infill.npz",
    "auk": "auk_2025_07_15-19_14_15_infill.npz",
    "bluejay": "bluejay_2025_07_15-19_14_15_infill.npz",
}


WITH_PARENT_SCORES = {
    "Auk": "subg_feat_eval_subgenre_with_parent_2025_09_17-19_56_25.npz",
    "BlueJay": "subg_feat_eval_subgenre_with_parent_2025_09_16-18_58_57.npz",
    "Crow d87": "crow-a-r87_subg_feat_eval_subgenre_with_parent_2025_09_21-00_00_50.npz",
    "Crow d87+2": "crow-d87+2_subg_feat_eval_subgenre_with_parent_2025_09_22-15_46_55.npz",
    # "Crow d89": "crow-d89_subg_feat_eval_subgenre_with_parent_2025_09_22-13_58_22.npz",
    # "Crow d89+4": "crow-d89+4_subg_feat_eval_subgenre_with_parent_2025_09_22-14_47_09.npz",
    # "Crow d101": "crow-a-r11_subg_feat_eval_subgenre_with_parent_2025_09_20-21_25_45.npz",
    # "Crow d102": "crow-a-r19_subg_feat_eval_subgenre_with_parent_2025_09_20-22_16_38.npz",
}

WITHIN_PARENT_SCORES = {
    "Auk": "subg_feat_eval_subgenre_within_parent_2025_09_17-19_56_25.npz",
    "BlueJay": "subg_feat_eval_subgenre_within_parent_2025_09_16-18_58_57.npz",
    "Crow d87": "crow-a-r87_subg_feat_eval_subgenre_within_parent_2025_09_21-00_00_50.npz",
    "Crow d87+2": "crow-d87+2_subg_feat_eval_subgenre_within_parent_2025_09_22-15_46_55.npz",
    # "Crow d89": "crow-d89_subg_feat_eval_subgenre_within_parent_2025_09_22-13_58_22.npz",
    # "Crow d89+4": "crow-d89+4_subg_feat_eval_subgenre_within_parent_2025_09_22-14_47_09.npz",
    # "Crow d101": "crow-a-r11_subg_feat_eval_subgenre_within_parent_2025_09_20-21_25_45.npz",
    # "Crow d102": "crow-a-r19_subg_feat_eval_subgenre_within_parent_2025_09_20-22_16_38.npz",
}

COVER_FILES = {
    "Training Data": "training_data_cover_self_sim.npz",
    "Auk": "ag-cov_feat_eval_cover_chirp-auk-t1_self_sim_2025_09_17-19_56_25.npz",
    "BlueJay": "ag-cov_feat_eval_cover_chirp-bluejay-t2_self_sim_2025_09_16-18_58_57.npz",
    "Crow d87": "crow-a-r87_ag-cov_feat_eval_cover_chirp-crow-eval_self_sim_2025_09_21-00_00_50.npz",
    "Crow d87+2": "crow-d87+2_ag-cov_feat_eval_cover_chirp-crow-eval_self_sim_2025_09_22-15_46_55.npz",
    # "Crow d89": "crow-d89_ag-cov_feat_eval_cover_chirp-crow-eval_self_sim_2025_09_22-13_58_22.npz",
    # "Crow d89+4": "crow-d89+4_ag-cov_feat_eval_cover_chirp-crow-eval_self_sim_2025_09_22-14_47_09.npz",
    # "Crow d101": "crow-a-r11_ag-cov_feat_eval_cover_chirp-crow-eval_self_sim_2025_09_20-21_25_45.npz",
    # "Crow d102": "crow-a-r19_ag-cov_feat_eval_cover_chirp-crow-eval_self_sim_2025_09_20-22_16_38.npz",
}

ARTIST_FILES = {
    "Training Data": "training_data_artist_artist_vox_sim.npz",
    "Auk": "ag-art_feat_eval_artist_chirp-auk-t1_artist_vox_sim_2025_09_17-19_56_25.npz",
    "BlueJay": "ag-art_feat_eval_artist_chirp-bluejay-t2_artist_vox_sim_2025_09_16-18_58_57.npz",
    "Crow d87": "crow-a-r87_ag-art_feat_eval_artist_chirp-crow-eval_artist_vox_sim_2025_09_21-00_00_50.npz",
    "Crow d87+2": "crow-d87+2_ag-art_feat_eval_artist_chirp-crow-eval_artist_vox_sim_2025_09_22-15_46_55.npz",
    # "Crow d89": "crow-d89_ag-art_feat_eval_artist_chirp-crow-eval_artist_vox_sim_2025_09_22-13_58_22.npz",
    # "Crow d89+4": "crow-d89+4_ag-art_feat_eval_artist_chirp-crow-eval_artist_vox_sim_2025_09_22-14_47_09.npz",
    # "Crow d101": "crow-a-r11_ag-art_feat_eval_artist_chirp-crow-eval_artist_vox_sim_2025_09_20-21_25_45.npz",
    # "Crow d102": "crow-a-r19_ag-art_feat_eval_artist_chirp-crow-eval_artist_vox_sim_2025_09_20-22_16_38.npz",
}

ARTIST_SIM_FILES = {}

ARTIST_SELF_SIM_FILES = {
    "Training Data": "training_data_artist_self_sim_120_30.npz",
    "Auk": "ag-art_feat_eval_artist_chirp-auk-t1_self_sim_2025_09_17-19_56_25.npz",
    "BlueJay": "ag-art_feat_eval_artist_chirp-bluejay-t2_self_sim_2025_09_16-18_58_57.npz",
    "Crow d87": "crow-a-r87_ag-art_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_21-00_00_50.npz",
    "Crow d87+2": "crow-d87+2_ag-art_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_22-15_46_55.npz",
    # "Crow d89": "crow-d89_ag-art_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_22-13_58_22.npz",
    # "Crow d89+4": "crow-d89+4_ag-art_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_22-14_47_09.npz",
    # "Crow d101": "crow-a-r11_ag-art_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_20-21_25_45.npz",
    # "Crow d102": "crow-a-r19_ag-art_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_20-22_16_38.npz",
}

ARTIST_HARD_SELF_SIM = {
    "Training Data": "training_data_artist_self_sim_120_30.npz",
    "Auk": "hard_feat_eval_artist_chirp-auk-t1_self_sim_2025_09_17-19_56_25.npz",
    "BlueJay": "hard_feat_eval_artist_chirp-bluejay-t2_self_sim_2025_09_16-18_58_57.npz",
    "Crow d87": "crow-a-r87_hard_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_21-00_00_50.npz",
    "Crow d87+2": "crow-d87+2_hard_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_22-15_46_55.npz",
    # "Crow d89": "crow-d89_hard_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_22-13_58_22.npz",
    # "Crow d89+4": "crow-d89+4_hard_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_22-14_47_09.npz",
    # "Crow d101": "crow-a-r11_hard_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_20-21_25_45.npz",
    # "Crow d102": "crow-a-r19_hard_feat_eval_artist_chirp-crow-eval_self_sim_2025_09_20-22_16_38.npz",
}