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

Unify API for challenges

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-01-31 20:48:05 -06:00
parent 82dfe20001
commit ccfcbe4e64
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
miplearn/problems/stab.py
View File

@@ -11,18 +11,24 @@ from scipy.stats import uniform, randint, bernoulli
from scipy.stats.distributions import rv_frozen from scipy.stats.distributions import rv_frozen
class MaxWeightStableSetChallengeA: class ChallengeA:
def __init__(self): def __init__(self,
seed=42,
n_training_instances=300,
n_test_instances=50,
):
np.random.seed(seed)
self.generator = MaxWeightStableSetGenerator(w=uniform(loc=100., scale=50.), self.generator = MaxWeightStableSetGenerator(w=uniform(loc=100., scale=50.),
n=randint(low=200, high=201), n=randint(low=200, high=201),
p=uniform(loc=0.05, scale=0.0), p=uniform(loc=0.05, scale=0.0),
fix_graph=True) fix_graph=True)
def get_training_instances(self): np.random.seed(seed + 1)
return self.generator.generate(300) self.training_instances = self.generator.generate(n_training_instances)
def get_test_instances(self): np.random.seed(seed + 2)
return self.generator.generate(50) self.test_instances = self.generator.generate(n_test_instances)
class MaxWeightStableSetGenerator: class MaxWeightStableSetGenerator: