Skip to content

Commit f157db0

Browse files
franchuteriveracharlesfu4
authored andcommitted
Removal of competition manager (automl#869)
* Removal of competition manager * Removed additional unused methods/files and moved metrics to estimator * Fix meta data generation * Make sure pytest is older newer than 4.6 * Unit tst fixing * flake8 fixes in examples * Fix metadata gen metrics
1 parent 0e33978 commit f157db0

File tree

2 files changed

+5
-7
lines changed

2 files changed

+5
-7
lines changed

autosklearn/automl.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1109,18 +1109,16 @@ def fit(
11091109
X, y,
11101110
X_test=X_test,
11111111
y_test=y_test,
1112-
task=task,
1113-
metric=metric,
1112+
task=REGRESSION,
11141113
feat_type=feat_type,
11151114
dataset_name=dataset_name,
11161115
only_return_configuration_space=only_return_configuration_space,
11171116
load_models=load_models,
11181117
)
11191118

1120-
def fit_ensemble(self, y, task=None, metric=None, precision='32',
1119+
def fit_ensemble(self, y, task=None, precision=32,
11211120
dataset_name=None, ensemble_nbest=None,
11221121
ensemble_size=None):
11231122
y = super()._check_y(y)
1124-
1125-
return super().fit_ensemble(y, task, metric, precision, dataset_name,
1126-
ensemble_nbest, ensemble_size)
1123+
return super().fit_ensemble(y, task, precision, dataset_name,
1124+
ensemble_nbest, ensemble_size)

test/test_automl/test_automl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -296,7 +296,7 @@ def test_automl_outputs(self):
296296
self.assertTrue(np.allclose(D.data['X_train'], X_train))
297297

298298
# Check that all directories are there
299-
fixture = ['cv_models', 'predictions_valid', 'true_targets_ensemble.npy',
299+
fixture = ['cv_models', 'true_targets_ensemble.npy',
300300
'start_time_100', 'datamanager.pkl',
301301
'predictions_ensemble',
302302
'ensembles', 'predictions_test', 'models']

0 commit comments

Comments
 (0)