Advertisement
Guest User

Untitled

a guest
Aug 21st, 2019
93
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.70 KB | None | 0 0
  1. def create_model():
  2.     '''Neural network with 3 hidden layers'''
  3.     model = Sequential()
  4.     model.add(Dense(256, input_dim=300, activation='relu', kernel_initializer='normal'))
  5.     model.add(Dropout(0.3))
  6.     model.add(Dense(256, activation='relu', kernel_initializer='normal'))
  7.     model.add(Dropout(0.5))
  8.     model.add(Dense(80, activation='relu', kernel_initializer='normal'))
  9.     model.add(Dense(2, activation="softmax", kernel_initializer='normal'))
  10.  
  11.     # gradient descent
  12.     sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
  13.    
  14.     # configure the learning process of the model
  15.     model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['accuracy'])
  16.     return model
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement