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

Merge components more efficiently

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-01-31 11:06:07 -06:00
parent 46e19eed1b
commit 2679522b76
4 changed files with 28 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
miplearn/warmstart.py
View File

@@ -180,13 +180,20 @@ class WarmStartComponent(Component):
self.predictors[category] = deepcopy(self.predictor_prototype)
self.predictors[category].fit(x_train, y_train)
def merge(self, other):
for c in other.x_train.keys():
if c not in self.x_train:
self.x_train[c] = other.x_train[c]
self.y_train[c] = other.y_train[c]
else:
self.x_train[c] = np.vstack([self.x_train[c], other.x_train[c]])
self.y_train[c] = np.vstack([self.y_train[c], other.y_train[c]])
if (c in other.predictors.keys()) and (c not in self.predictors.keys()):
self.predictors[c] = other.predictors[c]
def merge(self, other_components):
keys = set(self.x_train.keys())
for comp in other_components:
keys = keys.union(set(comp.x_train.keys()))
for key in keys:
x_train_submatrices = [comp.x_train[key]
for comp in other_components
if key in comp.x_train.keys()]
y_train_submatrices = [comp.y_train[key]
for comp in other_components
if key in comp.y_train.keys()]
if key in self.x_train.keys():
x_train_submatrices += [self.x_train[key]]
y_train_submatrices += [self.y_train[key]]
self.x_train[key] = np.vstack(x_train_submatrices)
self.y_train[key] = np.vstack(y_train_submatrices)