wehub-resource-sync
26 lines
534 B
Python
26 lines
534 B
Python
# coding:utf-8
|
|
import numpy as np
|
|
|
|
|
|
def one_hot(y):
|
|
n_values = np.max(y) + 1
|
|
return np.eye(n_values)[y]
|
|
|
|
|
|
def batch_iterator(X, batch_size=64):
|
|
"""Splits X into equal sized chunks."""
|
|
n_samples = X.shape[0]
|
|
n_batches = n_samples // batch_size
|
|
batch_end = 0
|
|
|
|
for b in range(n_batches):
|
|
batch_begin = b * batch_size
|
|
batch_end = batch_begin + batch_size
|
|
|
|
X_batch = X[batch_begin:batch_end]
|
|
|
|
yield X_batch
|
|
|
|
if n_batches * batch_size < n_samples:
|
|
yield X[batch_end:]
|