to_nparray.py

import itertools
import sys

import numpy as np
import yaml


with open(sys.argv[1]) as fp:
    data = yaml.load(fp)

labels = []
samples = []
for label, lsamples in data.items():
    labels.extend([label]*len(lsamples))
    samples.extend(lsamples)

labels = np.asarray(labels)
samples = np.asarray(samples)

assert labels.shape[0] == samples.shape[0]

with open(sys.argv[2] + "_samples.npz", "wb") as fp:
    np.savez(fp, samples)

with open(sys.argv[2] + "_labels.npz", "wb") as fp:
    np.savez(fp, labels)