def get_nn(name):
    models = {'simple_cnn': build_simple_cnn, 'simple_nn_for_training': build_simple_nn_for_training, 'hartmann': build_hartmann_network}
    return models[name]
def get_nn(name):
    models = {'simple_cnn': build_simple_cnn, 'simple_nn_for_training': build_simple_nn_for_training, 'hartmann': build_hartmann_network}
    return models[name]