generated from felixm/defaultpy
Fix sort order
This commit is contained in:
@@ -15,12 +15,12 @@ def get_sort_categories():
|
||||
return
|
||||
_, _, probs = learn.predict(row)
|
||||
cat_to_prob = dict(zip(learn.dls.vocab[1],probs.tolist()))
|
||||
categories.sort(key=lambda c: cat_to_prob[c] if c in cat_to_prob else 0.0)
|
||||
categories.sort(key=lambda c: cat_to_prob[c] if c in cat_to_prob else 0.0, reverse=True)
|
||||
|
||||
learn = None
|
||||
try:
|
||||
from fastai.text.all import load_learner
|
||||
learn = load_learner("ldg.pkl")
|
||||
learn = load_learner("export.pkl")
|
||||
except ModuleNotFoundError:
|
||||
pass
|
||||
except FileNotFoundError:
|
||||
|
||||
Reference in New Issue
Block a user