def simple_LSTM(): | |
np.random.seed(7) | |
model = Sequential(name = 'simple_LSTM') | |
model.add(LSTM(512, input_shape=(None, 6), recurrent_dropout=0.5)) | |
model.add(Dense(len(activities), activation='softmax')) | |
model.compile(loss='categorical_crossentropy', optimizer=Adam(lr=1e-4), metrics=['accuracy']) | |
print(model.summary()) | |
return model |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment