freemt commited on
Commit
d40a61e
·
1 Parent(s): 113657e

Update round(float(likelihood), 2) in ubee.py

Browse files
Files changed (1) hide show
  1. ubee/ubee.py +1 -0
ubee/ubee.py CHANGED
@@ -31,6 +31,7 @@ def ubee(
31
  for seq in sents_zh:
32
  ic(seq)
33
  label, likelihood = uclas(seq, labels, thresh=thresh)
 
34
  if label:
35
  res.append((seq, label, likelihood))
36
  try:
 
31
  for seq in sents_zh:
32
  ic(seq)
33
  label, likelihood = uclas(seq, labels, thresh=thresh)
34
+ likelihood = round(float(likelihood), 2)
35
  if label:
36
  res.append((seq, label, likelihood))
37
  try: