/
composite.py
82 lines (59 loc) · 2.52 KB
/
composite.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
# Copyright 2020 D-Wave Systems Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""A dimod composite that uses the steepest gradient descent to post-process
samples.
"""
import dimod
from dwave.samplers.greedy.sampler import SteepestDescentSolver
__all__ = ['SteepestDescentComposite']
class SteepestDescentComposite(dimod.ComposedSampler):
"""Runs greedy local optimization (steepest descent) on input problem,
seeded with samples from the sampler.
Args:
child_sampler (:class:`dimod.Sampler`):
A dimod sampler, such as a :class:`~dwave.system.samplers.DWaveSampler`.
"""
def __init__(self, child_sampler):
self.children = [child_sampler]
# set the parameters
self.parameters = child_sampler.parameters.copy()
# set the properties
self.properties = dict(child_properties=child_sampler.properties.copy())
parameters = None # overwritten by init
"""Parameters in the form of a dict.
For an instantiated composed sampler, keys are the keyword parameters
accepted by the child sampler and parameters added by the composite.
"""
children = None # overwritten by init
"""list [child_sampler]: List containing the structured sampler."""
properties = None # overwritten by init
"""Properties in the form of a dict.
Contains the properties of the child sampler.
"""
def sample(self, bqm: dimod.BinaryQuadraticModel, **parameters) -> dimod.SampleSet:
"""Find minima of a binary quadratic model.
Args:
bqm:
Binary quadratic model to be sampled from.
**parameters:
Parameters for the sampling method, specified by the child
sampler.
Returns:
:class:`dimod.SampleSet`.
"""
# solve the problem on the child system
child = self.child
sampleset = child.sample(bqm, **parameters)
greedy_sampler = SteepestDescentSolver()
return greedy_sampler.sample(bqm, initial_states=sampleset)