Add dirty_wellness_na to fill in NaN values

data_preparation/vectorization_ex.py

@@ -1,22 +1,12 @@
 import pandas as pd
 
 # read in CSV
-df = pd.read_csv('cleaned/dirty_wellness.csv')
+df = pd.read_csv('cleaned/dirty_wellness_na.csv')
 
 
 def vectorize_mult(column, dictionary, file=None):
-    """
-    Handles vectorizing
-    :param column:
-    :param dictionary:
-    :param file:
-    :return:
-    """
     newCol = column + "Num"
     df[newCol] = df[column].map(dictionary)
     if file is not None:
         df.to_csv('cleaned/{}.csv'.format(file))
 
-
-
-vectorize_mult("USGMeasurement", {"No": 0, "Yes": 1}, "wellness")