#pragma once
#include <string>
#include <vector>

struct FashionMNISTSet {
    // Images as floats in [0,1], row-major, size = N * 784
    std::vector<float> images;
    // Labels in {0..9}, size = N
    std::vector<uint8_t> labels;
    size_t N = 0;
};

// Loads the entire Fashion-MNIST dataset from the binary files
FashionMNISTSet load_fashion_mnist_data(const std::string& data_dir, bool is_train);