Re-run

fridrichmrtn · Sep 13, 2022 · 11e3e2e · 11e3e2e
1 parent 3db31bf
commit 11e3e2e
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/code/ingestion/run-ingestion.py b/code/ingestion/run-ingestion.py
@@ -24,31 +24,33 @@
 
 # CUSTOMER MODEL
 #_prerun_optimize_recom(dataset_name)
-# customer_model = construct_customer_model(
-#      dataset_name, time_steps=4, week_target=4)
-# save_customer_model(
-#      customer_model, dataset_name, overwrite=True)
+customer_model = construct_customer_model(
+     dataset_name, time_steps=4, week_target=4)
+save_customer_model(
+     customer_model, dataset_name, overwrite=True)
 
 # COMMAND ----------
 
 
 #
 ## REES46
 
-dataset_name = "rees46"
+#dataset_name = "rees46"
 
 # LOAD TRANSFORM
 #events = construct_events(dataset_name)
 #save_events(events, dataset_name)
 
 # CUSTOMER MODEL
 #_prerun_optimize_recom(dataset_name)
-customer_model = construct_customer_model(
-    dataset_name, time_steps=7, week_target=4)
-save_customer_model(customer_model,
-    dataset_name, overwrite=True)
+# customer_model = construct_customer_model(
+#     dataset_name, time_steps=7, week_target=4)
+# save_customer_model(customer_model,
+#     dataset_name, overwrite=True)
 
 # COMMAND ----------
 
-spark.table("churndb.rees46_customer_model").write.format("parquet")\
-    .mode("overwrite").save("dbfs:/mnt/rees46/raw/customer_model/")
+spark.table("churndb.retailrocket_customer_model").write.format("parquet")\
+    .mode("overwrite").save("dbfs:/mnt/retailrocket/raw/customer_model/")
+#spark.table("churndb.rees46_customer_model").write.format("parquet")\
+#    .mode("overwrite").save("dbfs:/mnt/rees46/raw/customer_model/")