Guest User

Untitled

a guest
Dec 16th, 2017
74
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.25 KB | None | 0 0
  1. user_id_input = Input(shape=[1], name='user')
  2. item_id_input = Input(shape=[1], name='item')
  3.  
  4. embedding_size = 10
  5. user_embedding = Embedding(output_dim=embedding_size, input_dim=users.shape[0],
  6. input_length=1, name='user_embedding')(user_id_input)
  7. item_embedding = Embedding(output_dim=embedding_size, input_dim=beers.shape[0],
  8. input_length=1, name='item_embedding')(item_id_input)
  9.  
  10. user_vecs = Reshape([embedding_size])(user_embedding)
  11. item_vecs = Reshape([embedding_size])(item_embedding)
  12.  
  13. # Add dense towers or not.
  14. # user_vecs = Dense(64, activation='relu')(user_vecs)
  15. # item_vecs = Dense(64, activation='relu')(item_vecs)
  16.  
  17. input_vecs = Multiply()([user_vecs, item_vecs]) # can be changed by concat or dot.
  18. # (if dot, no dense layer can be used after)
  19. input_vecs = Dropout(0.2)(input_vecs)
  20.  
  21. x = Dense(128, activation='relu')(input_vecs)
  22. #x = Dropout(0.2)(x) # Add droupout or not
  23. #x = Dense(64, activation='relu')(x) # Add dense again or not
  24. #x = Dropout(0.2)(x) # Add droupout or not
  25. #x = Dense(32, activation='relu')(x) # Add dense again or not
  26.  
  27. y = Dense(1)(x)
  28.  
  29. model = Model(inputs=[user_id_input, item_id_input], outputs=y)
  30. model.compile(optimizer='adam', loss='mse')
Add Comment
Please, Sign In to add comment