INNER CODE UNIT · Python
load_dataset
Tisha-runwal/Personalized-Federated-Learning-for-Privacy--Preserving-and-Scalable-IoT-Driven-Smart-Healthcare · pfl_hcare/fl/server.py:32
def load_dataset(dataset_name: str):
"""Load and return (train_dataset, test_dataset)."""
if dataset_name in ("har", "ucihar"):
from data.har_loader import HARDataset
train_ds = HARDataset(split="train", download=False)
test_ds = HARDataset(split="test", download=False)
else:
from data.mimic_loader import MedicalDataset
train_ds = MedicalDataset(split="train")
test_ds = MedicalDataset(split="test")
return train_ds, test_ds
# ---------------------------------------------------------------------------
# Local simulation (no Ray required)
# ---------------------------------------------------------------------------
def run_simulation(