update

utils/methods.py

@@ -60,6 +60,18 @@ def extractive_method(sentJson, sentFeat, model, threshold=0.5, TGB=False):
             true_proba[true_proba == np.max(true_proba)] = 1
         pred = (true_proba > threshold).astype('int')
         return pred, true_proba
+
+
+    grouped = sentFeat.groupby('section')
+    pred = []
+    true_proba = []
+
+    for group_name, group_data in grouped:
+        print(group_name)
+        pred_sec, true_proba_sec = predict(group_data)
+        pred.append(pred_sec)
+        true_proba.append(true_proba_sec)
+
     
     pred, true_proba = predict(sentFeat)
     body = convert_sentence_df(sentJson, pred, true_proba, TGB)