XGBoost integrated

[predictops.git] / predictops / engine.py

diff --git a/predictops/engine.py b/predictops/engine.py
index dedd2652cdd9efc97eaf24d26c1d3bd66a1398ec..f87e82e833fbd87bcc091f4dc568108a9bf86a21 100644 (file)
--- a/predictops/engine.py
+++ b/predictops/engine.py
@@ -7,6 +7,8 @@ from shutil import rmtree
 
 from .source.ephemeris import Ephemeris
 from .source.meteofrance import MeteoFrance
+from .learn.learning import Learning
+from .learn.preprocessing import Preprocessing
 from .target.target import Target
 
 fileConfig((Path.cwd() / 'config') / 'logging.cfg')
@@ -71,6 +73,18 @@ class Engine:
                               timestep = self._timestep)
 
 
+    def add_preprocessing(self):
+        self._preproc = Preprocessing(config_file = self._config,
+                                      dict_features = self.X,
+                                      dict_target = self.y)
+
+
+    def learn(self):
+        history = self._config['HISTORY_KNOWLEDGE'].getint('nb_lines')
+        self._learner = Learning(config_file = eval(self._config['LEARNER']['config']),
+                                 X = self._preproc.dataframe, y = list(self.y.values())[history:])
+
+
     @property
     def X(self):
         return self._X