get data from file

personas_vector/dssm_model.py

+import pandas as pd
+from utils.files import get_df
+from utils.utils import nth_element
+
+DEVICE_COLUMNS = [
+    "device_id",
+]
+
+TRACTATE_COLUMNS = [
+    "card_id",
+]
+
+
+def device_tractae_fe():
+    click_df = get_df("personas_tractate_click.csv")
+    exposure_df = get_df("personas_tractate_exposure.csv")
+    device_fe_df = get_df("personas_device_feature.csv")
+    tractate_tags_df = get_df("personas_tractate_tags.csv")
+    print(click_df.shape)
+    print(exposure_df.shape)
+    print(device_fe_df.shape)
+    print(tractate_tags_df.shape)
+
+
+if __name__ == "__main__":
+    click_df = get_df("personas_tractate_click.csv")
+    exposure_df = get_df("personas_tractate_exposure.csv")
+    device_fe_df = get_df("personas_device_feature.csv")
+    tractate_tags_df = get_df("personas_tractate_tags.csv")
+    print(click_df.shape)
+    print(exposure_df.shape)
+    print(device_fe_df.shape)
+    print(tractate_tags_df.shape)
+    print(tractate_tags_df.head(3))

personas_vector/personas_dssm_model.py

 import pandas as pd
+from utils.defs import nth_element
 from utils.files import get_df
-from utils.utils import nth_element
 
 DEVICE_COLUMNS = [
     "device_id",