Commit 138660cb authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Update fix random strategy (wip

parent 731cee0a
...@@ -70,6 +70,12 @@ def seed_job(seed_job_pb, seed, parameters, experiment_id, hyperparameters, verb ...@@ -70,6 +70,12 @@ def seed_job(seed_job_pb, seed, parameters, experiment_id, hyperparameters, verb
extraction_strategy=parameters['extraction_strategy'] extraction_strategy=parameters['extraction_strategy']
) )
pretrained_estimator = ModelFactory.build(dataset.task, pretrained_model_parameters, library=library) pretrained_estimator = ModelFactory.build(dataset.task, pretrained_model_parameters, library=library)
pretraned_trainer = Trainer(dataset)
pretraned_trainer.init(pretrained_estimator, subsets_used=parameters['subsets_used'])
pretrained_estimator.fit(
X=pretraned_trainer._X_forest,
y=pretraned_trainer._y_forest
)
else: else:
pretrained_estimator = None pretrained_estimator = None
pretrained_model_parameters = None pretrained_model_parameters = None
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment