eqc-models 0.11.1__py3-none-any.whl → 0.13.0__py3-none-any.whl

{eqc_models-0.11.1.data → eqc_models-0.13.0.data}/platlib/eqc_models/base/polynomial.py

@@ -1,10 +1,13 @@
 # (C) Quantum Computing Inc., 2024.
 from typing import Tuple, Union, List
+import logging
 import numpy as np
 from eqc_models.base.base import EqcModel
 from eqc_models.base.operators import Polynomial, QUBO, OperatorNotAvailableError
 from eqc_models.base.constraints import ConstraintsMixIn
 
+log = logging.getLogger(name=__name__)
+
 class PolynomialMixin:
     """This class provides an instance method and property that
     manage polynomial models.
@@ -46,7 +49,12 @@ class PolynomialMixin:
 
         """
 
-        value = self.polynomial.evaluate(np.array(solution))
+        coefficients, indices = self.H
+        log.debug("Coefficients (%d): %s", len(coefficients), coefficients)
+        log.debug("Indices (%d): %s", len(indices), indices)
+        polynomial = self.polynomial
+        log.debug("Polynomial: %s", polynomial)
+        value = polynomial.evaluate(np.array(solution))
 
         return value
     
@@ -107,8 +115,19 @@ class PolynomialModel(PolynomialMixin, EqcModel):
     """
 
     def __init__(self, coefficients : Union[List, np.ndarray], indices : Union[List, np.ndarray]) -> None:
+        # ensure that the coefficients are ordered and not duplicated
+        new_coefficients = {}
+        for idx, v in zip(indices, coefficients):
+            idx = tuple(idx)
+            if idx in new_coefficients:
+                new_coefficients[idx] += v
+            else:
+                new_coefficients[idx] = v
+        new_indices = [idx for idx in new_coefficients]
+        new_indices.sort()
+        coefficients = [new_coefficients[idx] for idx in new_indices]
         self.coefficients = coefficients
-        self.indices = indices
+        self.indices = new_indices
 
     @property
     def polynomial(self) -> Polynomial:

{eqc_models-0.11.1.data → eqc_models-0.13.0.data}/platlib/eqc_models/base/quadratic.py

@@ -245,6 +245,6 @@ class ConstrainedQuadraticModel(ConstraintsMixIn, QuadraticModel):
         return self.lhs, self.rhs
     
     def evaluateObjective(self, solution: np.ndarray) -> float:
-        J = self.quad_objective
-        C = self.linear_objective
+        J = np.array(self.quad_objective)
+        C = np.array(self.linear_objective)
         return np.squeeze(C.T @ solution + solution.T@J@solution)

{eqc_models-0.11.1.data → eqc_models-0.13.0.data}/platlib/eqc_models/base/results.py

@@ -1,4 +1,5 @@
 import dataclasses
+from typing import Dict
 import warnings
 import numpy as np
 
@@ -41,6 +42,14 @@ class SolutionResults:
     device : str
         String that represents the device used to solve the model.
 
+    raw_solutions : np.ndarray
+        Numpy array of the solutions as returned by the solver device.
+
+    calibration_time : float
+        Total time spend during job exectution where the device performed calibration.
+        The calibration is not directly affected by the job submission and the time
+        is not included in run_time.
+
     time_units : str
         String indicator of the unit of time reported in the metrics. Only
         ns is supported at this time.
@@ -56,6 +65,7 @@ class SolutionResults:
     postprocessing_time : np.ndarray
     penalties : np.ndarray = None
     device : str = None
+    raw_solutions : np.ndarray = None
     time_units : str = "ns"
 
     @property
@@ -95,7 +105,22 @@ class SolutionResults:
 
     @classmethod
     def from_cloud_response(cls, model, response, solver):
-        """ Fill in the details from the cloud """
+        """ 
+        Fill in the details from the cloud 
+
+        Parameters
+        ------------
+
+        model : eqc_models.base.EqcModel
+            EqcModel object describing the problem solved in response
+
+        response : Dict
+            Dictionary of the repsonse from the solver device.
+
+        solver : eqc_models.base.ModelSolver
+            ModelSolver object which is used to obtain job metrics.
+
+        """
 
         solutions = np.array(response["results"]["solutions"])
         if model.machine_slacks > 0:
@@ -164,3 +189,58 @@ class SolutionResults:
                                   device=device_type, time_units="ns")
 
         return results
+
+    @classmethod
+    def from_eqcdirect_response(cls, model, response, solver):
+        """
+        Fill in details from the response dictionary and possibly the solver device.
+
+        Parameters
+        ------------
+
+        model : eqc_models.base.EqcModel
+            EqcModel object describing the problem solved in response
+
+        response : Dict
+            Dictionary of the repsonse from the solver device.
+
+        solver : eqc_models.base.ModelSolver
+            ModelSolver object which is used to obtain device information. 
+
+        """
+        solutions = np.array(response["solution"])
+        if model.machine_slacks > 0:
+            solutions = solutions[:,:-model.machine_slacks]
+        energies = np.array(response["energy"])
+        # interrogate to determine the device type
+        info_dict = solver.client.system_info()
+        device_type = info_dict["device_type"]
+        if hasattr(model, "evaluateObjective"):
+            objectives = np.zeros((solutions.shape[0],), dtype=np.float32)
+            for i in range(solutions.shape[0]):
+                try:
+                    objective = model.evaluateObjective(solutions[i])
+                except NotImplementedError:
+                    warnings.warn(f"Cannot set objective value in results for {model.__class__}")
+                    objectives = None
+                    break
+                objectives[i] = objective
+        else:
+            objectives = None
+        if hasattr(model, "evaluatePenalties"):
+            penalties = np.zeros((solutions.shape[0],), dtype=np.float32)
+            for i in range(solutions.shape[0]):
+                penalties[i] = model.evaluatePenalties(solutions[i]) + model.offset
+        else:
+            penalties = None
+        counts = np.ones(solutions.shape[0])
+        runtime = response["runtime"]
+        post = response["postprocessing_time"]
+        pre = response["preprocessing_time"]
+        results = SolutionResults(solutions, energies, counts, objectives, 
+                                  runtime, pre, post, penalties=penalties,
+                                  device=device_type, time_units="s")
+
+        return results
+
+

eqc_models-0.13.0.data/platlib/eqc_models/graph/__init__.py

@@ -0,0 +1,11 @@
+# (C) Quantum Computing Inc., 2024.
+
+from .base import EdgeMixin, EdgeModel, GraphModel, NodeModel
+from .maxcut import MaxCutModel
+from .partition import GraphPartitionModel
+from .shortestpath import ShortestPathModel
+from .rcshortestpath import RCShortestPathModel
+
+__all__ = ["MaxCutModel", "GraphPartitionModel",
+           "EdgeMixin", "EdgeModel", "GraphModel", 
+           "NodeModel"]

{eqc_models-0.11.1.data → eqc_models-0.13.0.data}/platlib/eqc_models/graph/base.py

@@ -7,7 +7,7 @@ class GraphModel(QuadraticModel):
     """ """
     def __init__(self, G : nx.Graph):
         self.G = G
-        super().__init__(*self.costFunction())
+        super(GraphModel, self).__init__(*self.costFunction())
 
     @property
     def linear_objective(self):
@@ -63,12 +63,16 @@ class TwoPartitionModel(NodeModel):
     
     """
 
-class EdgeModel(GraphModel):
-    """ Create a model where the variables are edge-based """
+class EdgeMixin:
 
     @property
     def variables(self) -> List[str]:
         """ Provide a variable name to index lookup; order enforced by sorting the list before returning """
-        names = [f"({u},{v})" for u, v in self.G.edges]
+        names = [(u, v) for u, v in self.G.edges]
         names.sort()
         return names
+
+class EdgeModel(EdgeMixin, GraphModel):
+    """ Create a model where the variables are edge-based """
+
+    

eqc_models-0.13.0.data/platlib/eqc_models/graph/rcshortestpath.py

@@ -0,0 +1,81 @@
+import logging
+from typing import (Any, Dict, Tuple, List)
+import networkx as nx
+import numpy as np
+from eqc_models.graph.shortestpath import ShortestPathModel
+
+log = logging.getLogger(name=__name__)
+
+class RCShortestPathModel(ShortestPathModel):
+    """ 
+    Model for resource constrained shortest path problems.
+    Use a shortest path base model to implement the routing 
+    constraints and objective function. Add resource constraints.
+    
+    """
+
+    def __init__(self, G : nx.DiGraph, s : Any, t : Any, T : int, resource_key : str="resource"):
+        if T <= 0:
+            raise ValueError("T must be positive")
+        elif round(T, 0) != T:
+            raise ValueError("T must be integer-valued")
+        self.T = T
+        self.resource_key = resource_key
+        # determine shortest path by weight
+        nx_path = nx.shortest_path(G, s, t)
+        path_length = len(nx_path)
+        self.resource_mult = resource_mult = 2 * path_length / self.T
+        super(RCShortestPathModel, self).__init__(G, s, t)
+        upper_bound = np.ones(len(self.variables))
+        upper_bound[-1] = np.ceil(resource_mult * T)
+        self.upper_bound = upper_bound
+        is_discrete = [True for i in range(upper_bound.shape[0])]
+        is_discrete[-1] = False
+        self.is_discrete = is_discrete
+
+    @property
+    def variables(self):
+        variables = super(RCShortestPathModel, self).variables
+        return variables + ["resource_slack"]
+
+    def buildConstraints(self) -> Tuple[np.ndarray,np.ndarray]:
+        lhs, rhs = super(RCShortestPathModel, self).buildConstraints()
+        # add a single constraint
+        G = self.G
+        n = len(self.variables)
+        resource_lhs = np.zeros((1, n), dtype=np.float32)
+        resource_mult = self.resource_mult
+        log.debug("Resource multiplier %f", resource_mult)
+        for i in range(len(G.edges)):
+            (u, v) = self.variables[i]
+            # find the time to traverse the arc
+            resource_cost = G.edges[(u, v)][self.resource_key]
+            resource_cost *= resource_mult
+            log.debug(f"Adding resource %s for edge %s", resource_cost, (u, v))
+            resource_lhs[0, i] = resource_cost
+        resource_lhs[0, -1] = 1
+        lhs = self._stackLHS(lhs, resource_lhs)
+        rhs = np.hstack([rhs, [self.T*resource_mult]])
+        log.debug("LHS shape %s RHS shape %s", lhs.shape, rhs.shape)
+        return lhs, rhs
+
+    def pathCost(self, path):
+        """ sum the cost of all legs in the path """
+        assert self.s in path
+        G = self.G
+        node = self.s
+        max_len = len(self.G.nodes) - 1
+        path_len = 0
+        path_cost = 0
+        path_resources = 0
+        while node != t:
+            edge = (node, path[node])
+            if edge not in G.edges:
+                raise ValueError(f"Edge {edge} not found")
+            path_len += 1
+            path_cost += G.edges[edge]["weight"]
+            path_resources += G.edges[edge][self.resource_key]
+            if path_len > max_len:
+                raise ValueError("Invalid path. Describes a cycle.")
+        return path_cost, path_resources
+

eqc_models-0.13.0.data/platlib/eqc_models/graph/shortestpath.py

@@ -0,0 +1,190 @@
+r"""
+MIP Shortest Path implementation
+
+Given a graph $G$ and nodes $s$ and $t$, find the shortest path
+by edge weight between $s$ and $t$. 
+
+$$
+\min sum_ij w_ij x_ij
+$$
+subject to
+$$
+\sum_{(u,v)\in E} x_{u,v} - \sum{(v,u)\in E} x_{v,u} = 0 \forall u\in N\\{s,t}
+$$
+and
+$$
+\sum_{(s,v)\in E} x_{s,v} = 1
+$$
+and
+$$
+\sum_{(u,t)\in E} x_{u,t} = 1
+$$
+
+"""
+
+from typing import Any, Dict, Tuple
+import logging
+import numpy as np
+import networkx as nx
+from eqc_models.graph import EdgeMixin
+from eqc_models.base.quadratic import ConstrainedQuadraticModel
+
+log = logging.getLogger(name=__name__)
+
+class ShortestPathModel(EdgeMixin, ConstrainedQuadraticModel):
+    """
+    ShortestPathModel describes the MIP formulation for the 
+    shortest path problem.
+
+    Parameters
+    -------------
+
+    G : nx.DiGraph
+        A directed graph which is assumed to be connected. A graph
+        with disconnected subgraphs may reveal a solution if $s$ and $t$
+        are in the same subgraph, but testing for the existence of a path
+        between s and t using this model is not recommended. This is 
+        due to the difficulty posed by selecting a penalty multiplier 
+        large enough to enforce the panalties, which DNE in the infeasible 
+        case.
+    s : Any
+        This is the label for the start node.
+    t : Any
+        This is the label for the end node.
+
+
+    """
+
+    def __init__(self, G: nx.DiGraph, s : Any, t : Any):
+        self.G = G
+        self.s = s
+        self.t = t
+        self.lhs, self.rhs = self.buildConstraints()
+        C, J = self.buildObjective()
+        super(ShortestPathModel, self).__init__(C, J, self.lhs, self.rhs)
+        self.upper_bound = np.ones(self.lhs.shape[1])
+        self.is_discrete = [True for i in range(self.lhs.shape[1])]
+        self.machine_slacks = 0
+
+    def buildConstraints(self) -> Tuple[np.ndarray,np.ndarray]:
+        """
+        Constraints: 
+        $$
+        sum_j x[i,l] - sum_j x[j,l] = c for all l
+        $$
+        $c$ is -1, 1 or 0 for $i=t$, $s$ or all others
+        
+        """
+        log.debug("Building constraints to find path from %s to %s", self.s, self.t)
+        variables = self.variables
+        nodes = [n for n in self.G.nodes]
+        m = len(nodes)
+        n = len(variables)
+        _cons = np.zeros((m, n), dtype=np.int8)
+        _rhs = np.zeros((m, 1), dtype=np.int8)
+        for node_index, k in enumerate(nodes):
+            if k == self.s:
+                _rhs[node_index, 0] = 1
+            elif k == self.t:
+                _rhs[node_index, 0] = -1
+        for l in range(len(self.G.edges)): # don't enumerate the edges because the order could change
+            (i, j) = self.variables[l]
+            if i == j:
+                # self loops are not allowed
+                raise ValueError("Self loops are not allowed in ShortestPathModel")
+            # # ignore these edges because we can't go back to s or leave t
+            elif j == self.s:
+                continue
+            elif i == self.t:
+                continue
+            i_index = nodes.index(i)
+            j_index = nodes.index(j)
+            _cons[i_index, l] = 1
+            _cons[j_index, l] = -1
+        log.debug("LHS shape %s RHS shape %s", _cons.shape, _rhs.shape)
+        log.debug("checksum %f min %f", np.sum(_cons), np.min(_cons))
+        assert np.sum(_rhs) == 0
+        return _cons, np.squeeze(_rhs)
+
+    def buildObjective(self) -> Tuple[np.ndarray, np.ndarray]:
+        r"""
+        Objective:
+        $\min sum_ij w_ij x_ij$
+
+        """
+        variables = self.variables
+        G = self.G
+        nodes = G.nodes
+        m, n = len(nodes), len(variables)
+        _obj = [0 for i in range(n)]
+        for index, name in enumerate(variables):
+            if type(name) == type((1,2)):
+                i, j = name
+                _obj[index] = v = G.get_edge_data(i, j)["weight"]
+                assert not np.isnan(v), f"Got a NaN at {i, j}"
+        J = np.zeros((n, n))
+        return np.array(_obj), J
+
+    def decode(self, solution : np.ndarray) -> Dict:
+        """ 
+        Convert a solution to this model into a path, which is
+        a dictionary with each edge described by key, value pairs.
+
+        """
+        variables = self.variables[:len(self.G.edges)]
+        # log.debug("Using variables %s", variables)
+
+        lhs, rhs = self.constraints
+        log.debug("LHS shape %s RHS shape %s", lhs.shape, rhs.shape)
+        upper_thresh = max(solution[:len(variables)])
+        lower_thresh = 0
+        got_path = None
+        while upper_thresh - lower_thresh > 1e-6:
+            log.debug("Lower Value: %f Upper Value %f", lower_thresh, upper_thresh)
+            thresh = (lower_thresh + upper_thresh) / 2
+            nx_path = None
+            G = nx.DiGraph()
+            for (i, j), value in zip(variables, solution):
+                if value > thresh:
+                    G.add_edge(i, j)
+            edges = [e for e in G.edges]
+            log.debug("Resulting edge count %s", len(edges))
+            self.alt_g = G
+            path = {}
+            try:
+                nx_path = nx.shortest_path(G, self.s, self.t)
+                lower_thresh = thresh
+                got_path = nx_path
+                log.debug("Got path size %d", len(got_path))
+            except (nx.exception.NodeNotFound, nx.NetworkXAlgorithmError) as err:
+                upper_thresh = thresh
+        if got_path is None:
+            raise RuntimeError(f"Solution does not describe path from {self.s} to {self.t}")
+        # path = {}
+        # log.debug("Translating path to dictionary")
+        # for i, v in enumerate(got_path):
+        #     path[got_path[i-1]] = v
+        # log.debug("Updated path %s", path)
+        # if self.t in path:
+        #     log.debug("Removing %s from path keys.", self.t)
+        #     del path[self.t]
+        return got_path
+
+    def pathCost(self, path):
+        """ sum the cost of all legs in the path """
+        assert self.s in path
+        G = self.G
+        node = self.s
+        max_len = len(self.G.nodes) - 1
+        path_len = 0
+        path_cost = 0
+        while node != t:
+            edge = (node, path[node])
+            if edge not in G.edges:
+                raise ValueError(f"Edge {edge} not found")
+            path_len += 1
+            path_cost += G.edges[edge]["weight"]
+            if path_len > max_len:
+                raise ValueError("Invalid path. Describes a cycle.")
+        return path_cost
+

{eqc_models-0.11.1.data → eqc_models-0.13.0.data}/platlib/eqc_models/ml/classifierbase.py

@@ -11,6 +11,7 @@ import numpy as np
 
 from eqc_models import QuadraticModel
 from eqc_models.solvers.qciclient import Dirac3CloudSolver
+from eqc_models.solvers.eqcdirect import Dirac3DirectSolver
 
 
 class ClassifierBase(QuadraticModel):
@@ -18,12 +19,25 @@ class ClassifierBase(QuadraticModel):
         self,
         relaxation_schedule=2,
         num_samples=1,
+        solver_access="cloud",
+        ip_addr=None,
+        port=None,
     ):
 
         super(self).__init__(None, None, None)
         
         self.relaxation_schedule = relaxation_schedule
         self.num_samples = num_samples
+
+        assert solver_access in ["cloud", "direct"]
+
+        if solver_access == "direct":
+            assert ip_addr is not None, "ip_addr should be set when using direct solver!"
+            assert port is not None,"port should be set when using direct solver!"
+            
+        self.solver_access = solver_access
+        self.ip_addr = ip_addr
+        self.port = port
         self.params = None
         self.X_train = None
         self.y_train = None
@@ -53,18 +67,29 @@ class ClassifierBase(QuadraticModel):
         return
 
     def solve(self):
-        solver = Dirac3CloudSolver()
+
+        if self.solver_access == "direct":
+            solver = Dirac3DirectSolver()
+            solver.connect(self.ip_addr, self.port)
+        else:
+            solver = Dirac3CloudSolver()
+            
         response = solver.solve(
             self,
             sum_constraint=self._sum_constraint,
             relaxation_schedule=self.relaxation_schedule,
-            solution_precision=1,
             num_samples=self.num_samples,
         )
 
-        min_id = np.argmin(response["results"]["energies"])
-
-        sol = response["results"]["solutions"][min_id]
+        if self.solver_access == "cloud":
+            energies = response["results"]["energies"]
+            solutions = response["results"]["solutions"]
+        elif self.solver_access == "direct":
+            energies = response["energy"]
+            solutions = response["solution"]
+            
+        min_id = np.argmin(energies)
+        sol = solutions[min_id]
 
         print(response)
         

{eqc_models-0.11.1.data → eqc_models-0.13.0.data}/platlib/eqc_models/ml/classifierqboost.py

@@ -94,6 +94,12 @@ class QBoostClassifier(ClassifierBase):
 
     num_samples: Number of samples used by Dirac-3; default: 1.
 
+    solver_access: Solver access type: cloud or direct; default: cloud.
+
+    ip_addr: IP address of the device when direct access is used; default: None.
+    
+    port: Port number of the device when direct access is used; default: None.    
+    
     lambda_coef: A penalty multiplier; default: 0.
 
     weak_cls_schedule: Weak classifier schedule. Is either 1, 2,
@@ -155,6 +161,9 @@ class QBoostClassifier(ClassifierBase):
         self,
         relaxation_schedule=2,
         num_samples=1,
+        solver_access="cloud",
+        ip_addr=None,
+        port=None,
         lambda_coef=0,
         weak_cls_schedule=2,
         weak_cls_type="lg",
@@ -172,9 +181,13 @@ class QBoostClassifier(ClassifierBase):
             "multi_processing_shm",
             "sequential",
         ]
-
+        assert solver_access in ["cloud", "direct"]
+        
         self.relaxation_schedule = relaxation_schedule
         self.num_samples = num_samples
+        self.solver_access = solver_access
+        self.ip_addr = ip_addr
+        self.port = port
         self.lambda_coef = lambda_coef
         self.weak_cls_schedule = weak_cls_schedule
         self.weak_cls_type = weak_cls_type

{eqc_models-0.11.1.data → eqc_models-0.13.0.data}/platlib/eqc_models/ml/classifierqsvm.py

@@ -24,6 +24,12 @@ class QSVMClassifier(ClassifierBase):
 
     num_samples: Number of samples used by Dirac-3; default: 1.
 
+    solver_access: Solver access type: cloud or direct; default: cloud.
+
+    ip_addr: IP address of the device when direct access is used; default: None.
+    
+    port: Port number of the device when direct access is used; default: None.    
+    
     lambda_coef: The penalty multipler
 
     Examples
@@ -67,12 +73,20 @@ class QSVMClassifier(ClassifierBase):
         self,
         relaxation_schedule=1,
         num_samples=1,
+        solver_access="cloud",
+        ip_addr=None,
+        port=None,            
         lambda_coef=1.0,
     ):
         super(QSVMClassifier).__init__()
 
+        assert solver_access in ["cloud", "direct"]
+        
         self.relaxation_schedule = relaxation_schedule
         self.num_samples = num_samples
+        self.solver_access = solver_access
+        self.ip_addr = ip_addr
+        self.port = port        
         self.lambda_coef = lambda_coef
         self.fea_scaler = MinMaxScaler(feature_range=(-1, 1))
 
@@ -199,6 +213,12 @@ class QSVMClassifierDual(ClassifierBase):
 
     num_samples: Number of samples used by Dirac-3; default: 1.
 
+    solver_access: Solver access type: cloud or direct; default: cloud.
+
+    ip_addr: IP address of the device when direct access is used; default: None.
+    
+    port: Port number of the device when direct access is used; default: None.    
+    
     upper_limit: Coefficient upper limit; a regularization parameter;
     default: 1.0.
 
@@ -253,6 +273,9 @@ class QSVMClassifierDual(ClassifierBase):
         self,
         relaxation_schedule=2,
         num_samples=1,
+        solver_access="cloud",
+        ip_addr=None,
+        port=None,            
         upper_limit=1.0,
         gamma=1.0,
         eta=1.0,
@@ -260,8 +283,13 @@ class QSVMClassifierDual(ClassifierBase):
     ):
         super(QSVMClassifierDual).__init__()
 
+        assert solver_access in ["cloud", "direct"]
+        
         self.relaxation_schedule = relaxation_schedule
         self.num_samples = num_samples
+        self.solver_access = solver_access
+        self.ip_addr = ip_addr
+        self.port = port        
         self.upper_limit = upper_limit
         self.gamma = gamma
         self.eta = eta