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

removed bn output layer

Browse Source
This commit is contained in:
Malar Kannan
2017-12-11 14:12:23 +05:30
parent 05242d5991
commit 240ecb3f27
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
segment_model.py
View File

@@ -59,8 +59,7 @@ def simple_segment_model(input_dim):
d2 = Dense(8, activation='relu')(bn_d1)
bn_d2 = BatchNormalization(momentum=0.98)(d2)
d3 = Dense(1, activation='softmax')(bn_d2)
bn_d3 = BatchNormalization(momentum=0.98)(d3)
oup = Reshape(target_shape=(input_dim[0],))(bn_d3)
oup = Reshape(target_shape=(input_dim[0],))(d3)
return Model(inp, oup)
def write_model_arch(mod,mod_file):