Author: dligach
Date: Sat Sep 24 15:38:08 2016
New Revision: 1762132
URL: http://svn.apache.org/viewvc?rev=1762132&view=rev
Log:
need 5 (not 3) epochs to get best f1 for this model
Modified:
ctakes/trunk/ctakes-temporal/scripts/nn/hybrid_train.py
Modified: ctakes/trunk/ctakes-temporal/scripts/nn/hybrid_train.py
URL: http://svn.apache.org/viewvc/ctakes/trunk/ctakes-temporal/scripts/nn/hybrid_train.py?rev=1762132&r1=1762131&r2=1762132&view=diff
==============================================================================
--- ctakes/trunk/ctakes-temporal/scripts/nn/hybrid_train.py (original)
+++ ctakes/trunk/ctakes-temporal/scripts/nn/hybrid_train.py Sat Sep 24 15:38:08 2016
@@ -101,7 +101,7 @@ def main(args):
metrics=['accuracy'])
model.fit(train_xs,
train_y,
- nb_epoch=3,
+ nb_epoch=5,
batch_size=50,
verbose=1,
validation_split=0.1)
|