config.py

# Directories
FEATURES_DIR_PATH = "./features"
WEIGHTS_DIR_PATH = "./weights"
METADATA_DIR_PATH = "./metadata"
RESULTS_DIR_PATH = "./results"
PLAYLISTS_DIR_PATH = "./playlists"

# File Paths
DATA_PATH = "./MusAV"
DISCOGS_EMBEDDINGS_PATH = "./features/discogs_embeddings.pkl"
MSD_EMBEDDINGS_PATH = "./features/msd_embeddings.pkl"
GENRE_DISCOGS_PATH = "./features/genre_discogs.pkl"
ALL_FEATURES_PATH = "./features/all_features.pkl"
FILE_PATHS_PATH = "./features/file_paths.pkl"

# Result Paths
GENRE_COUNTS_PATH = "./results/all_styles.tsv"

# Playlist Paths
DESCRIPTOR_PLAYLIST_PATH = "./playlists/descriptor_playlist.m3u8"

# Model paths
TEMPOCNN_MODEL_PATH = "./weights/deepsquare-k16-3.pb"
DISCOGS_EMBEDDINGS_MODEL_PATH = "./weights/discogs-effnet-bs64-1.pb"
MSD_EMBEDDINGS_MODEL_PATH = "./weights/msd-musicnn-1.pb"
GENRE_DISCOGS_MODEL_PATH = "./weights/genre_discogs400-discogs-effnet-1.pb"
VOICE_DISCOGS_MODEL_PATH = "./weights/voice_instrumental-discogs-effnet-1.pb"
DANCEABILITY_DISCOGS_MODEL_PATH = "./weights/danceability-discogs-effnet-1.pb"
AROUSAL_MUSICNN_MODEL_PATH = "./weights/emomusic-msd-musicnn-2.pb"

# Metadata paths
DISCOGS_EMBEDDINGS_METADATA_PATH = "./metadata/discogs-effnet-bs64-1.json"
MSD_EMBEDDINGS_METADATA_PATH = "./metadata/msd-musicnn-1.json"