-
Notifications
You must be signed in to change notification settings - Fork 1
/
utils.py
44 lines (35 loc) · 1.3 KB
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import numpy as np
def sigmoid(x):
"""Computes the element wise logistic sigmoid of x.
Inputs:
x: Either a row vector or a column vector.
"""
return 1.0 / (1.0 + np.exp(-x))
def load_train():
"""Loads training data."""
with open('mnist_train.npz', 'rb') as f:
train_set = np.load(f)
train_inputs = train_set['train_inputs']
train_targets = train_set['train_targets']
return train_inputs, train_targets
def load_train_small():
"""Loads small training data."""
with open('mnist_train_small.npz', 'rb') as f:
train_set_small = np.load(f)
train_inputs_small = train_set_small['train_inputs_small']
train_targets_small = train_set_small['train_targets_small']
return train_inputs_small, train_targets_small
def load_valid():
"""Loads validation data."""
with open('mnist_valid.npz', 'rb') as f:
valid_set = np.load(f)
valid_inputs = valid_set['valid_inputs']
valid_targets = valid_set['valid_targets']
return valid_inputs, valid_targets
def load_test():
"""Loads test data."""
with open('mnist_test.npz', 'rb') as f:
test_set = np.load(f)
test_inputs = test_set['test_inputs']
test_targets = test_set['test_targets']
return test_inputs, test_targets