Merge pull request #5 from adevinta/feature/fix_na_random_forest

fix: Avoid NA in daily disaggregation
adevinta · May 21, 2024 · f045007 · f045007
2 parents ff4627c + ad44525
commit f045007
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/forecastout/core.py b/forecastout/core.py
@@ -157,7 +157,9 @@ def __get_daily_disaggregation_shares(self):
         )
         self.log.info(">> 3.3. Train disaggregation models")
         disaggregation_model_trainer = DisaggregationModelTrainer(
-            df_train=df_train_test.loc[df_train_test['test'] == 0].copy(),
+            df_train=(
+                df_train_test.loc[df_train_test['test'] == 0].dropna().copy()
+             ),
             config=self.config
         )
         disaggregation_model_trained = disaggregation_model_trainer.train(