Fix sort order

This commit is contained in:
2023-07-01 13:17:39 +02:00
parent a494db9b31
commit c96752e599

View File

@@ -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: