From 3d039650840ac4e5d4a5ea355c2bc26d371255ab Mon Sep 17 00:00:00 2001
From: arevelle-ensae <alexis.revelle@ensae.fr>
Date: Wed, 20 Mar 2024 12:07:24 +0000
Subject: [PATCH] commit segmentation

---
 0_6_Segmentation.py   | 18 ++++++++++++++++++
 utils_segmentation.py | 20 ++++++++++++++++++++
 2 files changed, 38 insertions(+)
 create mode 100644 0_6_Segmentation.py
 create mode 100644 utils_segmentation.py

diff --git a/0_6_Segmentation.py b/0_6_Segmentation.py
new file mode 100644
index 0000000..19ea3da
--- /dev/null
+++ b/0_6_Segmentation.py
@@ -0,0 +1,18 @@
+import pandas as pd
+import numpy as np
+import os
+import io
+import s3fs
+import re
+import pickle
+import warnings
+
+
+exec(open('utils_segmentation.py').read())
+warnings.filterwarnings('ignore')
+
+# Load Model 
+
+
+
+model = load_model(type_of_activity, model)
\ No newline at end of file
diff --git a/utils_segmentation.py b/utils_segmentation.py
new file mode 100644
index 0000000..56b3e1f
--- /dev/null
+++ b/utils_segmentation.py
@@ -0,0 +1,20 @@
+import pandas as pd
+import numpy as np
+import os
+import io
+import s3fs
+import re
+import pickle
+import warnings
+
+
+def load_model(type_of_activity, model):
+    BUCKET = f"projet-bdc2324-team1/Output_model/{type_of_activity}/{model}/"
+    filename = model + '.pkl'
+    file_path = BUCKET + filename
+    with fs.open(file_path, mode="rb") as f:
+        model_bytes = f.read()
+
+    model = pickle.loads(model_bytes)
+    return model
+