Guest User

Untitled

a guest
Mar 24th, 2018
107
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.25 KB | None | 0 0
  1. #### MLP
  2.  
  3. # Input variables
  4. mlp_user_input = Input(shape=(1,), dtype='int32', name = 'user_input')
  5. mlp_item_input = Input(shape=(1,), dtype='int32', name = 'item_input')
  6.  
  7. MLP_Embedding_User = Embedding(input_dim = n_users, output_dim = latent_dim, name = 'user_embedding', input_length=1,embeddings_regularizer = regularizers.l2(1e-5))
  8. MLP_Embedding_Item = Embedding(input_dim = n_prods, output_dim = latent_dim, name = 'item_embedding', input_length=1,embeddings_regularizer = regularizers.l2(1e-5))
  9.  
  10. # Crucial to flatten an embedding vector!
  11. mlp_user_latent = Flatten()(MLP_Embedding_User(mlp_user_input))
  12. mlp_item_latent = Flatten()(MLP_Embedding_Item(mlp_item_input))
  13.  
  14. # The 0-th layer is the concatenation of embedding layers
  15. vector = merge([mlp_user_latent, mlp_item_latent], mode = 'mul')
  16.  
  17. num_layer = 3
  18. layers = [200,100,50]
  19. # MLP layers
  20. for idx in range(1, num_layer):
  21. layer = Dense(layers[idx], kernel_regularizer=regularizers.l2(0.01), activation='relu',activity_regularizer=regularizers.l1(0.01), name = 'layer%d' %idx)
  22. vector = layer(vector)
  23.  
  24. # Final prediction layer
  25. mlp_prediction = Dense(1, activation='sigmoid', init='lecun_uniform', name = 'prediction')(vector)
  26. MLP = Model(input=[mlp_user_input, mlp_item_input], output=mlp_prediction)
Add Comment
Please, Sign In to add comment