def buildModel(self):
'''
:Description:
Build neural network model
'''
self.model = Sequential()
self.model.add(Embedding(self.embedding, 16, input_length=self.max_len))
for l in range(self.layers-1):
self.model.add(GRU(self.neurons, activation=self.activation, return_sequences=True, dropout_W=self.dropout, dropout_U=self.dropout))
self.model.add(GRU(self.neurons, activation=self.activation, return_sequences=False, dropout_W=self.dropout, dropout_U=self.dropout))
self.model.add(Dense(self.n_songs))
self.model.add(Activation('softmax'))
评论列表
文章目录