Spaces:

mickey1976
/

cove-api

Runtime error

mickey1976

Deploy: Minimal FastAPI backend for CoVE Space

549c270 about 1 year ago

417 Bytes

# 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

	# 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