You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.2 KiB
44 lines
1.2 KiB
# MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
|
|
# Copyright (C) 2020-2021, UChicago Argonne, LLC. All rights reserved.
|
|
# Released under the modified BSD license. See COPYING.md for more details.
|
|
|
|
import logging
|
|
from typing import Optional
|
|
|
|
from overrides import overrides
|
|
from pyomo import environ as pe
|
|
from scipy.stats import randint
|
|
|
|
from miplearn.solvers.pyomo.base import BasePyomoSolver
|
|
from miplearn.types import SolverParams
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class XpressPyomoSolver(BasePyomoSolver):
|
|
"""
|
|
An InternalSolver that uses XPRESS and the Pyomo modeling language.
|
|
|
|
Parameters
|
|
----------
|
|
params: dict
|
|
Dictionary of options to pass to the Pyomo solver. For example,
|
|
{"Threads": 4} to set the number of threads.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
params: Optional[SolverParams] = None,
|
|
) -> None:
|
|
if params is None:
|
|
params = {}
|
|
params["randomseed"] = randint(low=0, high=1000).rvs()
|
|
super().__init__(
|
|
solver_factory=pe.SolverFactory("xpress_persistent"),
|
|
params=params,
|
|
)
|
|
|
|
@overrides
|
|
def clone(self) -> "XpressPyomoSolver":
|
|
return XpressPyomoSolver(params=self.params)
|