Skip to content
This repository was archived by the owner on Jul 26, 2025. It is now read-only.

Commit 033dbb2

Browse files
committed
detect
1 parent 70bd45a commit 033dbb2

6 files changed

+18
-0
lines changed

brnn_sequence_analyzer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -573,6 +573,9 @@ def run(hidden_len=512, batch_size=128, nb_epoch=50, nb_iterations=4, lr=0.001,
573573
step=step, mapping=mapping)
574574
except KeyboardInterrupt:
575575
brnn.save_model("weights-stop.hdf5")
576+
elif mode == 'detect':
577+
detect(val_sequence, input_len, brnn, mapping=mapping,
578+
sentence_length=sentence_length)
576579
else:
577580
print "The mode = %s is not correct!!!" %mode
578581

brnn_sequence_analyzer_gen.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -592,6 +592,9 @@ def run(hidden_len=512, batch_size=128, nb_batch=200, nb_epoch=50,
592592
nb_epoch=nb_epoch, nb_iterations=nb_iterations)
593593
except KeyboardInterrupt:
594594
brnn.save_model("weights-stop.hdf5")
595+
elif mode == 'detect':
596+
detect(val_sequence, input_len, brnn, mapping=mapping,
597+
sentence_length=sentence_length)
595598
else:
596599
print "The mode = %s is not correct!!!" %mode
597600

rnn_sequence_analyzer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -549,6 +549,9 @@ def run(hidden_len=512, batch_size=128, nb_epoch=50, nb_iterations=4, lr=0.001,
549549
step=step, mapping=mapping)
550550
except KeyboardInterrupt:
551551
rnn.save_model("weights-stop.hdf5")
552+
elif mode == 'detect':
553+
detect(val_sequence, input_len, rnn, mapping=mapping,
554+
sentence_length=sentence_length)
552555
else:
553556
print "The mode = %s is not correct!!!" %mode
554557

rnn_sequence_analyzer_gen.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -570,6 +570,9 @@ def run(hidden_len=512, batch_size=128, nb_batch=200, nb_epoch=50,
570570
nb_epoch=nb_epoch, nb_iterations=nb_iterations)
571571
except KeyboardInterrupt:
572572
rnn.save_model("weights-stop.hdf5")
573+
elif mode == 'detect':
574+
detect(val_sequence, input_len, rnn, mapping=mapping,
575+
sentence_length=sentence_length)
573576
else:
574577
print "The mode = %s is not correct!!!" %mode
575578

sequence_analyzer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -681,6 +681,9 @@ def run(hidden_len=512, batch_size=128, nb_epoch=50, nb_iterations=4, lr=0.001,
681681
step=step, mapping=mapping)
682682
except KeyboardInterrupt:
683683
analyzer.save_model("weights-stop.hdf5")
684+
elif mode == 'detect':
685+
detect(val_sequence, input_len, analyzer, mapping=mapping,
686+
sentence_length=sentence_length)
684687
else:
685688
print "The mode = %s is not correct!!!" %mode
686689

sequence_analyzer_gen.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -699,6 +699,9 @@ def run(hidden_len=512, batch_size=128, nb_batch=200, nb_epoch=50,
699699
nb_epoch=nb_epoch, nb_iterations=nb_iterations)
700700
except KeyboardInterrupt:
701701
analyzer.save_model("weights-stop.hdf5")
702+
elif mode == 'detect':
703+
detect(val_sequence, input_len, analyzer, mapping=mapping,
704+
sentence_length=sentence_length)
702705
else:
703706
print "The mode = %s is not correct!!!" %mode
704707

0 commit comments

Comments
 (0)