#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ Created on Wed Mar 27 11:50:03 2019 @author: Macrobull """ import sys import numpy as np from collections import OrderedDict as Dict fn = sys.argv[1] input_names = sys.argv[2].split(':') output_name = sys.argv[3].split(':') data = np.load(fn) input_data = data['inputs'] output_data = data['outputs'] inputs = Dict(zip(input_names, [input_data])) outputs = Dict(zip(output_name, [output_data]))
np.savez(fn, inputs=inputs, outputs=outputs) # overwrite
