From f0bfd705aed2ca6ea7b49ad009502840eeafd4fd Mon Sep 17 00:00:00 2001 From: alirezade Date: Wed, 31 Jan 2024 11:12:53 +0100 Subject: [PATCH] fix memory leak error --- src/ProcessAnnotator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ProcessAnnotator.py b/src/ProcessAnnotator.py index e01b9a3..71897bb 100644 --- a/src/ProcessAnnotator.py +++ b/src/ProcessAnnotator.py @@ -94,7 +94,7 @@ def process(self, flow): def __predict_label(self, flow): df = pd.DataFrame(columns=flow.parameters.keys(), data=[flow.parameters.values()]) df.replace('', np.nan, inplace=True) - # y_pred = self.ml_model.predict_proba(df, verbose=0) + y_pred = self.ml_model.predict_proba(df, verbose=0) # y_pred_classes = np.argmax(y_pred, axis=0) # flow.add_parameter(Config.Texts.Prediction, self.label_index[str(y_pred_classes)]) # flow.add_parameter(Config.Texts.prediction_confidence, str(y_pred[y_pred_classes]))