src/data/loader.py

# src/data/loader.py

import json
from pathlib import Path

def load_dataset(dataset: str):
    base_path = Path("data/processed") / dataset

    with open(base_path / "seq.json", "r") as f:
        user_seqs = json.load(f)  # this keeps the full {user_id: [item_id, ...]} dict

    with open(base_path / "candidate_items.json", "r") as f:
        candidate_items = json.load(f)

    return user_seqs, candidate_items