Make tf-mhcflurry.py actually fit a model

openvax · Aug 1, 2016 · 95d8a3f · 95d8a3f
1 parent 14e4fb0
commit 95d8a3f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/script/tf-mhcflurry.py b/script/tf-mhcflurry.py
@@ -13,4 +13,8 @@
 from keras.models import Sequential
 from keras.layers import Dense, Activation
 model = Sequential()
-model.add(Dense(input_dim=9*21, output_dim=1))
+model.add(Dense(input_dim=9*21, output_dim=1)) # TF backend segfaults here
+model.add(Activation("sigmoid"))
+model.compile(loss="mse", optimizer="rmsprop")
+
+model.fit(training_hotshot, df_kmers[1], nb_epoch=5, batch_size=1)