yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-11-12 08:04:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

sample_weight when training with full data

Browse Source
This commit is contained in:
Chi Wang (MSR) 2021-02-05 23:12:52 -08:00
parent 833d022cfc
commit f956e957d4
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
flaml/automl.py
View File

@ -146,7 +146,9 @@ class AutoMLState:
else: else:
sampled_X_train, sampled_y_train = concat(self.X_train, sampled_X_train, sampled_y_train = concat(self.X_train,
self.X_val), np.concatenate([self.y_train, self.y_val]) self.X_val), np.concatenate([self.y_train, self.y_val])
sampled_weight = self.fit_kwargs.get('sample_weight') weight = self.fit_kwargs.get('sample_weight')
if weight is not None:
sampled_weight = np.concatenate([weight, self.weight_val])
return sampled_X_train, sampled_y_train, sampled_weight return sampled_X_train, sampled_y_train, sampled_weight
def _compute_with_config_base(self, def _compute_with_config_base(self,