[WIP] Start merging steps into a class for a pipeline.

exploration/ex_ml_pipeline.py

@@ -148,3 +148,17 @@ lin_reg_proximity.score(
     df_full_data_daily_means[["freq_prox_near", "prop_prox_near"]],
     df_full_data_daily_means["PA"],
 )
+
+# %% [markdown]
+# # Merging these into a pipeline
+
+# %%
+from machine_learning import pipeline
+
+# %%
+ml_pipeline = pipeline.MachineLearningPipeline()
+
+# %%
+ml_pipeline.participants_usernames
+
+# %%

machine_learning/pipeline.py

@@ -0,0 +1,12 @@
+import datetime
+
+import participants.query_db
+
+
+class MachineLearningPipeline:
+    def __init__(self, participants_usernames=None):
+        if participants_usernames is None:
+            participants_usernames = participants.query_db.get_usernames(
+                collection_start=datetime.date.fromisoformat("2020-08-01")
+            )
+        self.participants_usernames = participants_usernames