final updates

rulo96z · Dec 2, 2023 · 16430bb · 16430bb
1 parent 524a03d
commit 16430bb
Show file tree

Hide file tree

Showing 8 changed files with 9,858 additions and 8,786 deletions.
diff --git a/Pipfile b/Pipfile
@@ -20,6 +20,9 @@ python-dotenv = "*"
 scikit-learn = "*"
 scipy = "*"
 altair-viewer = "*"
+imbalanced-learn = "*"
+xgboost = "*"
+matplotlib = "*"
 
 [dev-packages]
 

diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/app/data.py b/app/data.py
@@ -79,7 +79,7 @@ def dataframe(self) -> DataFrame:
         df = DataFrame(data)
 
         # Drop '_id' column for visualization purposes
-        df = df.drop("_id", axis=1)
+        df = df.drop("_id", axis=1) # self.collection.find({}, {“_id”: False})
         return df
 
 

diff --git a/app/experimental_files/experiment_visualization.ipynb b/app/experimental_files/experiment_visualization.ipynb