fuhsiao commited on
Commit
ef27b68
·
1 Parent(s): f805bf4
Files changed (1) hide show
  1. utils/methods.py +1 -0
utils/methods.py CHANGED
@@ -50,6 +50,7 @@ def convert_sentence_df(sentJson, pred, true_proba, set_trigram_blocking):
50
  def extractive_method(sentJson, sentFeat, model, threshold=0.5, TGB=False):
51
  #預測
52
  def predict(x):
 
53
  true_proba = model.predict_proba(x)[:, 1]
54
  print(threshold)
55
  print(type(threshold))
 
50
  def extractive_method(sentJson, sentFeat, model, threshold=0.5, TGB=False):
51
  #預測
52
  def predict(x):
53
+ print(x)
54
  true_proba = model.predict_proba(x)[:, 1]
55
  print(threshold)
56
  print(type(threshold))