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

18
miplearn/problems/stab.py
View File

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