update null checker

src/models/esmm/fe/click_fe.py

@@ -14,9 +14,11 @@ def click_feature_engineering(click_df, conversion_df):
     cc_df.drop(["partition_date_x", "partition_date_y"], axis=1, inplace=True)
     cc_df["conversion_label"].fillna(0, inplace=True)
 
-    print("click:")
+    print("click: " + str(cc_df.shape))
     nullseries = cc_df.isnull().sum()
-    print(nullseries[nullseries > 0])
-    print(cc_df.shape)
+    nulls = nullseries[nullseries > 0]
+    if nulls.any():
+        print(nulls)
+        print("!!!!!!!!!!!!!!!!!!!!!!\n")
 
     return cc_df

src/models/esmm/fe/device_fe.py

@@ -87,7 +87,9 @@ def device_feature_engineering(device_df, content_type):
         columns = TRACTATE_DEVICE_COLUMNS
 
     nullseries = df.isnull().sum()
-    print("device:")
-    print(nullseries[nullseries > 0])
-    print(df.shape)
+    print("device: " + str(df.shape))
+    nulls = nullseries[nullseries > 0]
+    if nulls.any():
+        print(nulls)
+        print("!!!!!!!!!!!!!!!!!!!!!!\n")
     return df[columns]

src/models/esmm/fe/diary_fe.py

@@ -85,10 +85,12 @@ def diary_feature_engineering(df):
 
     diary_df = diary_df[DIARY_COLUMNS]
 
-    print("diary:")
+    print("diary: " + str(diary_df.shape))
     nullseries = diary_df.isnull().sum()
-    print(nullseries[nullseries > 0])
-    print(diary_df.shape)
+    nulls = nullseries[nullseries > 0]
+    if nulls.any():
+        print(nulls)
+        print("!!!!!!!!!!!!!!!!!!!!!!\n")
     return diary_df
 
 

src/models/esmm/fe/tractate_fe.py

@@ -86,10 +86,12 @@ def tractate_feature_engineering(tractate_df):
 
     df = df[TRACTATE_COLUMNS]
 
-    print("tractate:")
+    print("tractate: " + str(df.shape))
     nullseries = df.isnull().sum()
-    print(nullseries[nullseries > 0])
-    print(df.shape)
+    nulls = nullseries[nullseries > 0]
+    if nulls.any():
+        print(nulls)
+        print("!!!!!!!!!!!!!!!!!!!!!!\n")
     return df