ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 09:28:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Implement BenchmarkRunner

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-01-23 21:59:59 -06:00
parent 07090bac9e
commit 8f141e6a9d
10 changed files with 161 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
miplearn/problems/stab.py
View File

@@ -15,10 +15,12 @@ class MaxStableSetGenerator:
self.base_weights = base_weights
self.perturbation_scale = perturbation_scale
def generate(self):
perturbation = np.random.rand(self.graph.number_of_nodes()) * self.perturbation_scale
weights = self.base_weights + perturbation
return MaxStableSetInstance(self.graph, weights)
def generate(self, n_samples):
def _sample():
perturbation = np.random.rand(self.graph.number_of_nodes()) * self.perturbation_scale
weights = self.base_weights + perturbation
return MaxStableSetInstance(self.graph, weights)
return [_sample() for _ in range(n_samples)]
class MaxStableSetInstance(Instance):