From f73f86486a77d110e8a59b57f3fe11287a47f3d4 Mon Sep 17 00:00:00 2001
From: junos <junos.lukan@ijs.si>
Date: Sat, 21 Aug 2021 18:28:22 +0200
Subject: [PATCH] Fill communication features with appropriate values.

---
 machine_learning/pipeline.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/machine_learning/pipeline.py b/machine_learning/pipeline.py
index e07adae..efc4cb1 100644
--- a/machine_learning/pipeline.py
+++ b/machine_learning/pipeline.py
@@ -75,6 +75,11 @@ class SensorFeatures:
             self.df_features_all = safe_outer_merge_on_index(
                 self.df_features_all, self.df_calls_sms
             )
+            self.df_features_all.fillna(
+                value=communication.FILL_NA_CALLS_SMS_ALL,
+                inplace=True,
+                downcast="infer",
+            )
 
     def get_features(self, data_type, feature_names) -> pd.DataFrame:
         if data_type == "proximity":