sarapy 1.1.5__tar.gz → 1.2.6__tar.gz

{sarapy-1.1.5/sarapy.egg-info → sarapy-1.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sarapy
-Version: 1.1.5
+Version: 1.2.6
 Home-page: https://github.com/lucasbaldezzari/sarapy
 Author: Lucas Baldezzari
 Author-email: Lucas Baldezzari <lmbaldezzari@gmail.com>
@@ -19,6 +19,14 @@ Requires-Dist: geopy
 
 Library for processing SARAPICO project metadata of _AMG SA_.
 
+#### Version 1.2.6
+
+- Se agrega línea dentro de clase OpsProcessor.
+
+#### Version 1.1.6
+
+- Se modifica PlantinClassifier para mejorar el NO conteo de plantines en OSP.
+
 #### Version 1.1.5
 
 - Se quita print de función update.

{sarapy-1.1.5 → sarapy-1.2.6}/README.md

@@ -2,6 +2,14 @@
 
 Library for processing SARAPICO project metadata of _AMG SA_.
 
+#### Version 1.2.6
+
+- Se agrega línea dentro de clase OpsProcessor.
+
+#### Version 1.1.6
+
+- Se modifica PlantinClassifier para mejorar el NO conteo de plantines en OSP.
+
 #### Version 1.1.5
 
 - Se quita print de función update.

{sarapy-1.1.5 → sarapy-1.2.6}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "sarapy"
 
-version = "1.1.5"
+version = "1.2.6"
 authors = [
   {name = "Lucas Baldezzari", email = "lmbaldezzari@gmail.com"},]
 maintainers = [

{sarapy-1.1.5 → sarapy-1.2.6}/sarapy/dataProcessing/OpsProcessor.py

@@ -185,7 +185,7 @@ class OpsProcessor():
             ##clasificamos las operaciones para plantín
             operations = self.operationsDict[ID_NPDP]["sample_ops"]
             features, dst_pt, inest_pt = self.plantinFMCreator.fit_transform(operations)
-            classified_ops = self._plantin_classifier.classify(features, **classify_kwargs)
+            classified_ops = self._plantin_classifier.classify(features, dst_pt, inest_pt, **classify_kwargs)
             
             ##chequeo si first_day_op_classified es True, si es así, no se considera la primera fila de las classified_ops
             if self.operationsDict[ID_NPDP]["first_day_op_classified"]:

{sarapy-1.1.5 → sarapy-1.2.6}/sarapy/mlProcessors/PlantinClassifier.py

@@ -23,14 +23,18 @@ class PlantinClassifier(BaseEstimator, TransformerMixin):
         except FileNotFoundError:
             print("El archivo no se encuentra en el directorio actual.")
         
-    def classify(self, feature_matrix, update_samePlace:bool = True, **kwargs):
+    def classify(self, feature_matrix, dst_pt, inest_pt,
+                update_samePlace:bool = True, update_dstpt: bool = True,
+                umbral_proba = 0.85, **kwargs):
         """Genera la clasificación de las operaciones para plantines.
         
-        feature_matrix: Es un array con los datos (strings) provenientes de la base de datos histórica.
+        - feature_matrix: Es un array con los datos (strings) provenientes de la base de datos histórica.
         La forma de newData debe ser (n,3). Las columnas de newData deben ser,
                 - 1: deltaO
                 - 2: ratio_dCdP
                 - 3: distancias
+        - dst_pt: Array con las distorsiones de plantín.
+        - inest_pt: Array con flag de inestabilidad de plantín.
 
         kwargs: Diccionario con los argumentos necesarios para la clasificación.
 
@@ -38,11 +42,15 @@ class PlantinClassifier(BaseEstimator, TransformerMixin):
         """
 
         self.clasificaiones = self._pipeline.predict(feature_matrix)
+        self.probas = self._pipeline.predict_proba(feature_matrix)
 
         if update_samePlace:
             self.grouped_ops = self.groupOpsSamePlace(feature_matrix, **kwargs)
             self.clasificaiones = self.updateLabelsSamePlace(self.clasificaiones, self.grouped_ops)
 
+        if update_dstpt:
+            self.clasificaiones = self.updateLabelsFromDSTPT(self.clasificaiones, dst_pt, inest_pt, umbral_proba)
+
         return self.clasificaiones
     
     def groupOpsSamePlace(self, X, useRatioStats = True, std_weight=1, useDistancesStats = True,
@@ -116,6 +124,20 @@ class PlantinClassifier(BaseEstimator, TransformerMixin):
 
         return new_labels
     
+    def updateLabelsFromDSTPT(self, labels, dst_pt, inest_pt, umbral_proba = 0.85):
+        """
+        Función para actualizar las etiquetas de las operaciones que tengan distorsiones de plantín.
+        """
+        new_labels = labels.copy()
+        ##filtro si dst_pt es menor a 7 y si inest_pt es 0
+        new_labels[(dst_pt < 7) & (inest_pt == 0)] = 0
+
+        ##si inest_pt es y umbral_proba es menor a umbra_proba, entonces la operación es 0
+        new_labels[(inest_pt == 1) & (self.probas[:,1] < umbral_proba)] = 0
+        # new_labels[(inest_pt == 1) & (self.probas[:,1] >= umbral_proba)] = 1
+
+        return new_labels
+    
 if __name__ == "__main__":
     import os
     import pandas as pd
@@ -128,8 +150,8 @@ if __name__ == "__main__":
     fmcreator = PlantinFMCreator.PlantinFMCreator(imputeDistances=False)
     tindata = TransformInputData.TransformInputData()
 
-    data_path = os.path.join(os.getcwd(), "examples\\2024-09-04\\UPM011N\\data.json")
-    historical_data_path = os.path.join(os.getcwd(), "examples\\2024-09-04\\UPM011N\\historical-data.json")
+    data_path = os.path.join(os.getcwd(), "examples\\2024-10-15\\UPM015N\\data.json")
+    historical_data_path = os.path.join(os.getcwd(), "examples\\2024-10-15\\UPM015N\\historical-data.json")
     raw_data = pd.read_json(data_path, orient="records").to_dict(orient="records")
     raw_data2 = pd.read_json(historical_data_path, orient="records").to_dict(orient="records")
 
@@ -141,5 +163,6 @@ if __name__ == "__main__":
     rf_clf_nu = PlantinClassifier.PlantinClassifier(classifier_file='modelos\\pipeline_rf.pkl') ##wu = no update
     rf_clf_wu = PlantinClassifier.PlantinClassifier(classifier_file='modelos\\pipeline_rf.pkl') ##wu = with update
 
-    print(rf_clf_nu.classify(X, update_samePlace = False).mean())
-    print(rf_clf_wu.classify(X, update_samePlace=True, useRatioStats=True, useDistancesStats=True).mean())
+    print(rf_clf_nu.classify(X, dst_pt, inest_pt, update_samePlace = False, update_dstpt=False).mean())
+    print(rf_clf_wu.classify(X, dst_pt, inest_pt, update_samePlace=True, update_dstpt=True,
+    useRatioStats=True, useDistancesStats=True,umbral_proba=0.8).mean()) 

{sarapy-1.1.5 → sarapy-1.2.6}/sarapy/version.py

@@ -1,2 +1,2 @@
 ## Version of the package   
-__version__ = "1.1.5"
+__version__ = "1.2.6"

{sarapy-1.1.5 → sarapy-1.2.6/sarapy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sarapy
-Version: 1.1.5
+Version: 1.2.6
 Home-page: https://github.com/lucasbaldezzari/sarapy
 Author: Lucas Baldezzari
 Author-email: Lucas Baldezzari <lmbaldezzari@gmail.com>
@@ -19,6 +19,14 @@ Requires-Dist: geopy
 
 Library for processing SARAPICO project metadata of _AMG SA_.
 
+#### Version 1.2.6
+
+- Se agrega línea dentro de clase OpsProcessor.
+
+#### Version 1.1.6
+
+- Se modifica PlantinClassifier para mejorar el NO conteo de plantines en OSP.
+
 #### Version 1.1.5
 
 - Se quita print de función update.