Advertisement
Guest User

Untitled

a guest
Jan 22nd, 2019
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.02 KB | None | 0 0
  1. def create_model():
  2. """
  3. This method create neural network model.
  4. :return: model
  5. """
  6.  
  7. model = Sequential()
  8. model.add(Convolution2D(16, (5, 5), input_shape=(64, 64, 6)))
  9. model.add(BatchNormalization())
  10. model.add(Activation('relu'))
  11. model.add(MaxPooling2D(pool_size=(2, 2)))
  12.  
  13. model.add(Convolution2D(64, (5, 5)))
  14. model.add(BatchNormalization())
  15. model.add(Activation('relu'))
  16. model.add(MaxPooling2D(pool_size=(2, 2)))
  17.  
  18. model.add(Convolution2D(256, (5, 5)))
  19. model.add(BatchNormalization())
  20. model.add(Activation('relu'))
  21.  
  22. model.add(Flatten())
  23. model.add(Dropout(0.5))
  24.  
  25. model.add(Dense(640))
  26. model.add(BatchNormalization())
  27. model.add(Activation('relu'))
  28.  
  29. model.add(Dropout(0.5))
  30. model.add(Dense(2))
  31. model.add(Activation('softmax'))
  32.  
  33. sgd = SGD(lr=0.0001, momentum=0.9, decay=0.005)
  34. model.compile(optimizer='sgd', loss="categorical_crossentropy", metrics=['accuracy'])
  35.  
  36. print(model.summary())
  37. return model
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement