Skip to content
Extraits de code Groupes Projets
Valider 9ec41711 rédigé par Adrien Payen's avatar Adrien Payen
Parcourir les fichiers

import constants

parent 4fc1e638
Aucune branche associée trouvée
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
# third parties imports
from pathlib import Path
class Constant:
"""Constants for dataset paths and column names."""
DATA_PATH = Path('data/tiny') # -- fill here the dataset size to use
# Content
CONTENT_PATH = DATA_PATH / 'content' # Path to content data
# - item
ITEMS_FILENAME = 'movies.csv' # Filename for items data
ITEM_ID_COL = 'movieId' # Column name for item IDs
LABEL_COL = 'title' # Column name for item labels
GENRES_COL = 'genres' # Column name for item genres
# Evidence
EVIDENCE_PATH = DATA_PATH / 'evidence' # Path to evidence data
# - ratings
RATINGS_FILENAME = 'ratings.csv' # Filename for ratings data
USER_ID_COL = 'userId' # Column name for user IDs
RATING_COL = 'rating' # Column name for ratings
TIMESTAMP_COL = 'timestamp' # Column name for timestamps
USER_ITEM_RATINGS = [USER_ID_COL, ITEM_ID_COL, RATING_COL] # List of columns for user-item ratings
EVALUATION_PATH = DATA_PATH / 'evaluations' # Path to evaluation data
# Rating scale
RATINGS_SCALE = (0.5, 5.0) # -- fill in here the ratings scale as a tuple (min_value, max_value)
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Veuillez vous inscrire ou vous pour commenter