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

Make get_instance_features return np.ndarray

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-09 14:02:14 -05:00
parent 47d3011808
commit 56b39b6c9c
6 changed files with 29 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/problems/knapsack.py
View File

@@ -94,8 +94,8 @@ class MultiKnapsackInstance(Instance):
return model
@overrides
def get_instance_features(self) -> List[float]:
return [float(np.mean(self.prices))] + list(self.capacities)
def get_instance_features(self) -> np.ndarray:
return np.array([float(np.mean(self.prices))] + list(self.capacities))
@overrides
def get_variable_features(self) -> Dict[str, List[float]]: