add prediction and change training and test

[asr1617data.git] / test.py

diff --git a/test.py b/test.py
index 72a787c..7378596 100644 (file)
--- a/test.py
+++ b/test.py
@@ -13,7 +13,7 @@ model.compile(
 
 model.summary()
 
-dat = np.genfromtxt('test.txt', dtype=float, delimiter='\t')[:, range(1, 14)]
-lab = np.genfromtxt('test.txt', dtype=int, delimiter='\t')[:, 0]
+dat = np.genfromtxt('test.txt', dtype=float, delimiter='\t', usecols=range(1, 14))
+lab = np.genfromtxt('test.txt', dtype=int, delimiter='\t', usecols=[0])
 
 print(model.evaluate(dat, lab, batch_size=32))