Advertisement
Guest User

Untitled

a guest
Jan 17th, 2017
77
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.09 KB | None | 0 0
  1. model = Sequential()
  2. model.add(Convolution2D(nb_filters, kernel_size[0], kernel_size[1],
  3. border_mode='valid',
  4. input_shape=input_shape))
  5. model.add(Activation('relu'))
  6. model.add(Convolution2D(nb_filters, kernel_size[0], kernel_size[1]))
  7. model.add(Activation('relu'))
  8. model.add(MaxPooling2D(pool_size=pool_size))
  9. # (16, 16, 32)
  10. model.add(Convolution2D(nb_filters*2, kernel_size[0], kernel_size[1]))
  11. model.add(Activation('relu'))
  12. model.add(Convolution2D(nb_filters*2, kernel_size[0], kernel_size[1]))
  13. model.add(Activation('relu'))
  14. model.add(MaxPooling2D(pool_size=pool_size))
  15. # (8, 8, 64) = (2048)
  16. model.add(Flatten())
  17. model.add(Dense(1024))
  18. model.add(Activation('relu'))
  19. model.add(Dropout(0.5))
  20. model.add(Dense(2)) # define a binary classification problem
  21. model.add(Activation('softmax'))
  22.  
  23. model.compile(loss='categorical_crossentropy',
  24. optimizer='adadelta',
  25. metrics=['accuracy'])
  26. model.fit(x_train, y_train,
  27. batch_size=batch_size,
  28. nb_epoch=nb_epoch,
  29. verbose=1,
  30. validation_data=(x_test, y_test))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement