src/dataset.py

import json
from datasets import load_dataset

def load_emotion_dataset():
    dataset = load_dataset("json", data_files={
        "train": "data/train.jsonl",
        "validation": "data/valid.jsonl",
        "test": "data/test.jsonl"
    })
    return dataset