Added show_missing and drop_missing methods

Author: tirthajyoti

utils/NN_trainer.py

@@ -5,10 +5,6 @@ class Trainer(object):
     Requirements: Numpy, Pandas, Matplotlib, Scikit-learn, Keras (TensorFlow)
     """
 
-    import numpy as np
-    import matplotlib.pyplot as plt
-    import pandas as pd
-
     def __init__(self):
         """
         Initialization
@@ -113,6 +109,23 @@ def drop_static_cols(self):
             df2 = df1
         self.df = df2
 
+    def show_missing(self):
+        """
+        Shows how many missing values are there
+        """
+        print("The following table shows the number of missing values in the dataset (by the features)\n")
+        print(self.df.isna().sum())
+
+    def drop_missing(self):
+        """
+        Drops rows of data which have missing values.
+        This method does not return a DataFrame but modifies the internal DataFrame.
+        """
+        df1 = self.df
+        df2 = df1.dropna(axis=0)
+
+        self.df = df2
+
     def set_outputs(self, output_cols=None):
         """
         Sets the list the output columns (to be modeled)
@@ -550,7 +563,11 @@ def model_in_plain_english(self):
             print("No model has been built yet!")
             return None
 
-        d = self.model.get_config()['layers']
+        config = self.model.get_config()
+        if type(config)==dict:
+            d = config['layers']
+        else:
+            d = config
         num_layers = len(d)
         total_param = self.model.count_params()
         print("Here is the model description")