Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from keras.activations import relu, elu, linear, sigmoid
- params = {'lr': [0.001, 0.1],
- 'first_neuron':[5,9],
- 'hidden_layers':[1,5,10],
- 'batch_size': [30,40,50],
- 'epochs': [40],
- 'dropout': [0,0.2],
- 'kernel_initializer': ['normal'],
- 'optimizer': [Adam],
- #'loss':[mean_absolute_error],
- 'activation':[],
- 'last_activation':['linear']
- }
- regression__model = regression_model(X_air_train, y_air_train, X_air_valid, y_air_valid, params)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement