Skip to content

Instantly share code, notes, and snippets.

@mikkokotila
Last active May 18, 2018 18:07
Show Gist options
  • Save mikkokotila/b4d386d898b41a6da3373290391a2d8d to your computer and use it in GitHub Desktop.
Save mikkokotila/b4d386d898b41a6da3373290391a2d8d to your computer and use it in GitHub Desktop.
breast cancer dataset example params
# then we can go ahead and set the parameter space
p = {'lr': (0.5, 5, 10),
'first_neuron':[4, 8, 16, 32, 64],
'hidden_layers':[0, 1, 2],
'batch_size': (2, 30, 10),
'epochs': [150],
'dropout': (0, 0.5, 5),
'weight_regulizer':[None],
'emb_output_dims': [None],
'shape':['brick','long_funnel'],
'optimizer': [Adam, Nadam, RMSprop],
'losses': [logcosh, binary_crossentropy],
'activation':[relu, elu],
'last_activation': [sigmoid]}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment