malar
/
speech-scoring
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

using the full data and fixed typo in model layer name

master
Malar Kannan 2017-12-11 13:47:30 +05:30
parent a6543491f8
commit fea9184aec
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
segment_model.py
View File

@ -54,7 +54,7 @@ def simple_segment_model(input_dim):
b_gr1 = Bidirectional(GRU(256, return_sequences=True),merge_mode='sum')(inp)
# b_gr1
# b_gr2 = Bidirectional(GRU(64, return_sequences=True),merge_mode='sum')(b_gr1)
b_gr3 = Bidirectional(GRU(64, return_sequences=True),merge_mode='sum')(b_gr2)
b_gr3 = Bidirectional(GRU(64, return_sequences=True),merge_mode='sum')(b_gr1)
d1 = Dense(32, activation='relu')(b_gr3)
d2 = Dense(8, activation='relu')(d1)
d3 = Dense(1, activation='softmax')(d2)
@ -131,5 +131,5 @@ def train_segment(collection_name = 'test',resume_weights='',initial_epoch=0):
if __name__ == '__main__':
pass
# train_segment('story_phrases','./models/segment/story_phrases.1000/speech_segment_model-final.h5',1001)
# pass
train_segment('story_phrases_full')#,'./models/segment/story_phrases.1000/speech_segment_model-final.h5',1001)