YOLO_MATH/yolo_split_data/data.yaml

# data.yaml (Save this next to the 'yolo_split_data' folder)

# Root path where the split data is located
path: /home/dipesh/Downloads/api-mcq/YOLO_MATH/yolo_split_data

# Paths to the split directories (relative to the 'path' above)
train: images/train
val: images/val
test: images/test

# Number of classes
nc: 2

# Class names
names: ['equation', 'figure']