root 2 settimane fa
parent
commit
6d6d0facfe
1 ha cambiato i file con 3 aggiunte e 3 eliminazioni
  1. 3 3
      ai-slibra-assistant/util/utils.py

+ 3 - 3
ai-slibra-assistant/util/utils.py

@@ -205,8 +205,8 @@ def timetic(func):
 def loaddict():
     loc = dict()
     df = pd.read_excel("../data/location.xlsx", header=0)
-    loc['zh'] = dict(df[['norm_name', 'name']].values)
-    loc['pinyin'] = dict(df[['name_pinyin', 'name']].values)
+    loc['zh'] = dict(df[['name', 'norm_name']].values)
+    loc['pinyin'] = dict(df[['name_pinyin', 'norm_name']].values)
     short_val = [(i, 80) for i in df['short_name'].dropna().tolist()]
     norm_val = [(i, 100) for i in df['norm_name'].dropna().tolist()]
     norm_val.extend(short_val)
@@ -258,7 +258,7 @@ def norm_community(asr):
 
 
 if __name__ == "__main__":
-    asr="佳木斯长青乡四核村"
+    asr="滨江小区"
     print(f"asr:{asr}")
     loc = norm_community(asr)
     print(f"命中小区:{loc}")