Skip to content

Instantly share code, notes, and snippets.

@jbaiter
Created April 7, 2016 13:46
Show Gist options
  • Save jbaiter/b1c153b52f5223b87e49c1d9299b8d1b to your computer and use it in GitHub Desktop.
Save jbaiter/b1c153b52f5223b87e49c1d9299b8d1b to your computer and use it in GitHub Desktop.
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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment