preprocess.py

import numpy as np

"""This file contains some functions related to preprocessing."""

def onehot(output_labels):
	return np.squeeze((np.unique(output_labels) == output_labels[:, None]).astype(np.float32))

def get_fans(shape):
	return np.prod(shape[:-1]), shape[-1]

def xavier_init(shape):
	fan_in, fan_out = get_fans(shape)
	dev = np.sqrt(6.0 / (fan_in + fan_out))
	return np.random.uniform(-dev, dev, shape)

def xavier_init_normal(shape):
	fan_in, fan_out = get_fans(shape)
	dev = np.sqrt(3.0 / (fan_in + fan_out))
	return np.random.normal(size=shape, scale=dev)

def randomize(data, labels):
	order = np.arange(data.shape[0])
	np.random.shuffle(order)
	data = data[order]
	labels = labels[order]
	return data, labels