Guest User

Untitled

a guest
Feb 21st, 2018
73
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.13 KB | None | 0 0
  1. dimension = 120
  2. model_users = Sequential()
  3. model_users.add(Embedding(len(np.unique(users)), dimension))
  4. model_users.add(Reshape((dimension,)))
  5.  
  6. model_movies = Sequential()
  7. model_movies.add(Embedding(len(np.unique(movies)), dimension, input_length=1))
  8. model_movies.add(Reshape((dimension,)))
  9.  
  10. model = Sequential()
  11. model.add(Merge([model_users, model_movies], mode = 'concat'))
  12. model.add(Dropout(0.1))
  13. model.add(Dense(100, activation = 'relu'))
  14. model.add(Dropout(0.1))
  15. model.add(Dense(500, activation = 'sigmoid'))
  16. model.add(Dropout(0.1))
  17. model.add(Dense(dimension, activation = 'linear'))
  18. model.add(Dropout(0.1))
  19. model.add(Dense(5, activation = 'softmax'))
  20. model.compile(loss = 'categorical_crossentropy', optimizer = 'adam', metrics=['accuracy'])
  21. print(model.summary())
  22.  
  23. dimensions = len(movies_unique)
  24.  
  25. model = Sequential()
  26. model.add(Masking(mask_value = 0, input_shape = (dimensions, dimensions)))
  27. model.add(LSTM(32, return_sequences = True))
  28. model.add(TimeDistributed(Dense(len(ratings_unique), activation = 'relu')))
  29. model.add(Activation('softmax'))
  30. model.compile(loss = 'mse', optimizer = 'adam', metrics = ['accuracy'])
  31. print(model.summary())
Add Comment
Please, Sign In to add comment