Skip to content
Snippets Groups Projects
Commit 138660cb authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Update fix random strategy (wip

parent 731cee0a
Branches
No related tags found
2 merge requests!20Resolve "integration-sota",!19WIP: Resolve "Adding new datasets"
......@@ -70,6 +70,12 @@ def seed_job(seed_job_pb, seed, parameters, experiment_id, hyperparameters, verb
extraction_strategy=parameters['extraction_strategy']
)
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:
pretrained_estimator = None
pretrained_model_parameters = None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment