Skip to content

Instantly share code, notes, and snippets.

@shshemi
Last active August 28, 2019 10:35
Show Gist options
  • Save shshemi/67ba17c92ce6df69e628102bc5753fc0 to your computer and use it in GitHub Desktop.
Save shshemi/67ba17c92ce6df69e628102bc5753fc0 to your computer and use it in GitHub Desktop.
def get_model(item_count=5):
input_weights = Input((item_count,))
input_prices = Input((item_count,))
inputs_concat = Concatenate()([input_weights, input_prices])
picks = Dense(35, activation="sigmoid")(inputs_concat)
picks = Dense(item_count, activation="sigmoid")(picks)
model = Model(inputs=[input_weights, input_prices], outputs=[picks])
return model
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment