transform original int data into float

dapeng2018 · Oct 20, 2014 · 407dfae · 407dfae
1 parent 8af3811
commit 407dfae
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/data/data.py b/data/data.py
@@ -1,9 +1,10 @@
 import os
-
 import numpy as np
 
 curpath = os.path.dirname(__file__)
-x_train = np.load(curpath + '/x_train.npy')
-x_test = np.load(curpath + '/x_test.npy')
+x_train_int = np.load(curpath + '/x_train.npy')
+x_train = x_train_int / 255.0
+x_test_int = np.load(curpath + '/x_test.npy')
+x_test = x_test_int / 255.0
 y_train = np.load(curpath + '/y_train.npy')
 y_test = np.load(curpath + '/y_test.npy')