Skip to content

Instantly share code, notes, and snippets.

@crawftv
Created April 11, 2019 03:55
Show Gist options
  • Save crawftv/d1ec2e6c92fa6fee83495cc5ed5bae47 to your computer and use it in GitHub Desktop.
Save crawftv/d1ec2e6c92fa6fee83495cc5ed5bae47 to your computer and use it in GitHub Desktop.
Skopt Tutorial - Search Dimensions
dim_learning_rate = Real(low=1e-4, high=1e-2, prior='log-uniform',
name='learning_rate')
dim_num_dense_layers = Integer(low=1, high=5, name='num_dense_layers')
dim_num_input_nodes = Integer(low=1, high=512, name='num_input_nodes')
dim_num_dense_nodes = Integer(low=1, high=28, name='num_dense_nodes')
dim_activation = Categorical(categories=['relu', 'sigmoid'],
name='activation')
dim_batch_size = Integer(low=1, high=128, name='batch_size')
dim_adam_decay = Real(low=1e-6,high=1e-2,name="adam_decay")
dimensions = [dim_learning_rate,
dim_num_dense_layers,
dim_num_input_nodes,
dim_num_dense_nodes,
dim_activation,
dim_batch_size,
dim_adam_decay
]
default_parameters = [1e-3, 1,512, 13, 'relu',64, 1e-3]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment