add prediction and change training and test

[asr1617data.git] / train.py

diff --git a/train.py b/train.py
index a4ecbbc..2f6bc91 100644 (file)
--- a/train.py
+++ b/train.py
@@ -4,7 +4,7 @@ from keras.layers import Dense, Dropout  # , Activation
 
 model = Sequential()
 
-model.add(Dense(26, input_shape=(13,), activation='relu'))
+model.add(Dense(2000, input_shape=(13,), activation='relu'))
 model.add(Dense(1, activation='sigmoid'))
 
 model.compile(
@@ -14,8 +14,8 @@ model.compile(
 
 model.summary()
 
-dat = np.genfromtxt('train.txt', dtype=float, delimiter='\t')[:, range(1, 14)]
-lab = np.genfromtxt('train.txt', dtype=int, delimiter='\t')[:, 0]
+dat = np.genfromtxt('train.txt', dtype=float, delimiter='\t', usecols=range(1, 14))
+lab = np.genfromtxt('train.txt', dtype=int, delimiter='\t', usecols=[0])
 
 model.fit(dat, lab, epochs=10, batch_size=32)