From 138660cb2d778cdb41029fc77a3e7d7c8a29bf35 Mon Sep 17 00:00:00 2001
From: Charly Lamothe <charly.lamothe@univ-amu.fr>
Date: Fri, 6 Mar 2020 17:17:22 +0100
Subject: [PATCH] Update fix random strategy (wip
---
code/train.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/code/train.py b/code/train.py
index 72c91d8..e70902b 100644
--- a/code/train.py
+++ b/code/train.py
@@ -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
--
GitLab