Yasin Yousif commited on
Commit
c64789a
·
1 Parent(s): 8a8221e
Files changed (1) hide show
  1. story.py +2 -2
story.py CHANGED
@@ -31,6 +31,6 @@ def story_model(preds,res_len=5):
31
  for word,prob in preds:
32
  dists += np.array([prob * (word in sent) for sent in All_sentenses])
33
 
34
- f_res = list(zip(dists,All_sentenses)).sort(key=lambda x:x[0])[-res_len:]
35
 
36
- return [f for _,f in f_res] # list of sentenses
 
31
  for word,prob in preds:
32
  dists += np.array([prob * (word in sent) for sent in All_sentenses])
33
 
34
+ f_res = list(zip(dists,All_sentenses)).sort(key=lambda x:x[0])
35
 
36
+ return [f for _,f in f_res[-res_len:]] # list of sentenses