felixm/ledgerai
1
0
Fork 0
generated from felixm/defaultpy
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix sort order

Browse Source
This commit is contained in:
felixm
2023-07-01 13:17:39 +02:00
parent a494db9b31
commit c96752e599
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/predict.py
View File

@@ -15,12 +15,12 @@ def get_sort_categories():
return return
_, _, probs = learn.predict(row) _, _, probs = learn.predict(row)
cat_to_prob = dict(zip(learn.dls.vocab[1],probs.tolist())) 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 learn = None
try: try:
from fastai.text.all import load_learner from fastai.text.all import load_learner
learn = load_learner("ldg.pkl") learn = load_learner("export.pkl")
except ModuleNotFoundError: except ModuleNotFoundError:
pass pass
except FileNotFoundError: except FileNotFoundError: