update

utils/methods.py

@@ -51,6 +51,9 @@ def extractive_method(sentJson, sentFeat, model, threshold=0.5, TGB=False):
     #預測
     def predict(x):
         true_proba = model.predict_proba(x)[:, 1]
+        print(threshold)
+        print(type(threshold))
+        print(true_proba)
         # 如果沒有任何句子的預測機率大於閾值，則選取最大機率的句子為摘要句
         if not np.any(true_proba > threshold):
             true_proba[true_proba == np.max(true_proba)] = 1