Skip to content

Instantly share code, notes, and snippets.

@bkaankuguoglu

bkaankuguoglu/get_model

Created April 1, 2021 21:42
Show Gist options
  • Save bkaankuguoglu/f99ddffb3a05d20e1413496de3b4745a to your computer and use it in GitHub Desktop.
Save bkaankuguoglu/f99ddffb3a05d20e1413496de3b4745a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
get_model
def get_model(model, model_params):
models = {
"rnn": RNNModel,
"lstm": LSTMModel,
"gru": GRUModel,
}
return models.get(model.lower())(**model_params)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment