LessDataTest

app.py

@@ -74,68 +74,68 @@ TEST_PHRASES = [
     "Сегодня оплатил в телемаркет Крикова 655,80 лей",
     "Вчера был в такуме на 425,7 лей",
     "Сегодня купил в tagaer на 285,40 лей",
-    "Вчера оплатил в Supraten 1200,50 лень",
-    "Сегодня был в стоматологии о фамилии на 455,90 лень",
-    "Я на следующей неделе заказал в стартур билеты на 855,60 лень",
-    "Сегодня оплатил в Sankos 245,70 лей",
-    "Вчера купил в Samurai на 325,40 лей",
-    "Сегодня был в Salomer на 185,50 lei",
-    "Вчера купил vragob na 655,80 lei",
-    "Сегодня купил в primul discounter на 425,03 лей",
-    "Вчера оплатил Premier Energy 985,90 lei",
-    "Сегодня заказали в пицце 9 на 285,60 лей",
-    "На прошлой неделе ходили в piața centrală, купили на 455,7 lei",
-    "Сегодня купил в peon farm на 325,40 лей",
-    "Вчера Wallach купила в Peach Girl на 755,50 лей",
-    "Через 2 дня купил в Pandashop на 895,80 лей",
-    "Pazavchora był vivația i kupil na 185,30 lei",
-    "Сегодня оплатил в oldcom 655,90 лей",
-    "Вчера купил рыбу в Ocean Fish на 280 lei",
-    "Сегодня купил в номер 1 на 420 лей",
-    "вчера воля купила в nanu market на 250 lei",
-    "Сегодня купил в Mozza на 380 lei",
-    "Вчера оплатил moldpressa 90 lei",
-    "Сегодня заплатил в Moldova-Presa 180 lei",
-    "Вчера платил MoldovaGaz 1250 lei",
-    "Сегодня был в modus vivendi, я ставил 420 lei",
-    "Вчера купил в Micov na 150 lei",
-    "Сегодня оплатил в метрах 890,13 лей",
-    "Вчера купил в Мерси на 210 lei",
-    "Сегодня заплатил в Megapolis 680 lei",
-    "Вчера Оля купила лекарство в Medical Market на 340 лей",
-    "Сегодня оплатил в максимум 450 lei",
-    "Вчера купил десерт в maestro delice на 120 lei",
-    "Сегодня оплатил в maestro 750 lei",
-    "вчера оля купила в local на 190 лей",
-    "Сегодня был в Linelo 115 и купил на 280 лей",
-    "Вчера купил продукты в Linel на 420,55 лей",
-    "Сегодня оплатил vats 320 lei",
-    "Вчера Олечка была в Kiss Beauty Salon на 450 lei",
-    "Сегодня купил кебаб в кебаб на 150 лей",
-    "Вчера Оля была в Кауфленд и потратила 890,15 лей",
-    "Сегодня купил в cat shop на 650 lei",
-    "Вчера вечером был выкатан суши на 300 восьятлей",
-    "Оля вчера заказала в Joom на 1200 lei",
-    "Сегодня купили рассаду в Ярди Маркет на 280 лей",
-    "Вчера Влад оплатил в uiti credit 950 lei",
-    "Сегодня оплатил в U.T. Credit очередной платеж 1800 лей",
-    "Вчера заказал в iherb витамина на 420 лей",
-    "На прошлой неделе покупали в Ярмареку на 950,13 лей",
-    "Оля вчера была в Хипократис и оставила 650 lei",
-    "Сегодня я купил витамины в herb на 180 лей",
-    "Вчера купил хлеб в Граньер на 70 лей",
-    "Сегодня ходил в Global Store за техникой на 2100 лей",
-    "Вчера я оплатил интернет в Giganet 450,35 лей",
-    "Сегодня Оля купила хлеб Франзелуца на 80 петлей",
-    "вчера купил рыбу в эфилете на 420 лей",
-    "На прошлой неделе заплатил в Fidesco 1300 lei",
-    "Сегодня Влад был в Феличи и купил сыр на 95 лей",
-    "Вчера вечером купили fast food на 180 lei",
-    "Олечка вчера купила лекарство фармачия Familia на 240 лей",
-    "Я сегодня утром оплатил Eurotelicom 310 lei",
-    "Вчера Владислав оплатил энергоком 560 lei",
-    "Сегодня оплатил в Елика 420 лей",
-    "На следующей неделе в субботу хочу зайти в дёйнер-кебаб",
+    # "Вчера оплатил в Supraten 1200,50 лень",
+    # "Сегодня был в стоматологии о фамилии на 455,90 лень",
+    # "Я на следующей неделе заказал в стартур билеты на 855,60 лень",
+    # "Сегодня опл��тил в Sankos 245,70 лей",
+    # "Вчера купил в Samurai на 325,40 лей",
+    # "Сегодня был в Salomer на 185,50 lei",
+    # "Вчера купил vragob na 655,80 lei",
+    # "Сегодня купил в primul discounter на 425,03 лей",
+    # "Вчера оплатил Premier Energy 985,90 lei",
+    # "Сегодня заказали в пицце 9 на 285,60 лей",
+    # "На прошлой неделе ходили в piața centrală, купили на 455,7 lei",
+    # "Сегодня купил в peon farm на 325,40 лей",
+    # "Вчера Wallach купила в Peach Girl на 755,50 лей",
+    # "Через 2 дня купил в Pandashop на 895,80 лей",
+    # "Pazavchora był vivația i kupil na 185,30 lei",
+    # "Сегодня оплатил в oldcom 655,90 лей",
+    # "Вчера купил рыбу в Ocean Fish на 280 lei",
+    # "Сегодня купил в номер 1 на 420 лей",
+    # "вчера воля купила в nanu market на 250 lei",
+    # "Сегодня купил в Mozza на 380 lei",
+    # "Вчера оплатил moldpressa 90 lei",
+    # "Сегодня заплатил в Moldova-Presa 180 lei",
+    # "Вчера платил MoldovaGaz 1250 lei",
+    # "Сегодня был в modus vivendi, я ставил 420 lei",
+    # "Вчера купил в Micov na 150 lei",
+    # "Сегодня оплатил в метрах 890,13 лей",
+    # "Вчера купил в Мерси на 210 lei",
+    # "Сегодня заплатил в Megapolis 680 lei",
+    # "Вчера Оля купила лекарство в Medical Market на 340 лей",
+    # "Сегодня оплатил в максимум 450 lei",
+    # "Вчера купил десерт в maestro delice на 120 lei",
+    # "Сегодня оплатил в maestro 750 lei",
+    # "вчера оля купила в local на 190 лей",
+    # "Сегодня был в Linelo 115 и купил на 280 лей",
+    # "Вчера купил продукты в Linel на 420,55 лей",
+    # "Сегодня оплатил vats 320 lei",
+    # "Вчера Олечка была в Kiss Beauty Salon на 450 lei",
+    # "Сегодня купил кебаб в кебаб на 150 лей",
+    # "Вчера Оля была в Кауфленд и потратила 890,15 лей",
+    # "Сегодня купил в cat shop на 650 lei",
+    # "Вчера вечером был выкатан суши на 300 восьятлей",
+    # "Оля вчера заказала в Joom на 1200 lei",
+    # "Сегодня купили рассаду в Ярди Маркет на 280 лей",
+    # "Вчера Влад оплатил в uiti credit 950 lei",
+    # "Сегодня оплатил в U.T. Credit очередной платеж 1800 лей",
+    # "Вчера заказал в iherb витамина на 420 лей",
+    # "На прошлой неделе покупали в Ярмареку на 950,13 лей",
+    # "Оля вчера была в Хипократис и оставила 650 lei",
+    # "Сегодня я купил витамины в herb на 180 лей",
+    # "Вчера купил хлеб в Граньер на 70 лей",
+    # "Сегодня ходил в Global Store за техникой на 2100 лей",
+    # "Вчера я оплатил интернет в Giganet 450,35 лей",
+    # "Сегодня Оля купила хлеб Франзелуца на 80 петлей",
+    # "вчера купил рыбу в эфилете на 420 лей",
+    # "На прошлой неделе заплатил в Fidesco 1300 lei",
+    # "Сегодня Влад был в Феличи и купил сыр на 95 лей",
+    # "Вчера вечером купили fast food на 180 lei",
+    # "Олечка вчера купила лекарство фармачия Familia на 240 лей",
+    # "Я сегодня утром оплатил Eurotelicom 310 lei",
+    # "Вчера Владислав оплатил энергоком 560 lei",
+    # "Сегодня оплатил в Елика 420 лей",
+    # "На следующей неделе в субботу хочу зайти в дёйнер-кебаб",
 ]
 
 

extractors/user_extractor.py

@@ -60,7 +60,6 @@ class ExpenseUserExtractor:
         self.threshold = threshold
         self.supplier_terms = {normalize_text(supplier) for supplier in suppliers}
         self.user_terms = [normalize_text(user) for user in users]
-        self.user_lookup = dict(zip(self.user_terms, self.users))
         self.user_embeddings = model.encode(
             [f"passage: {user}" for user in self.user_terms],
             convert_to_tensor=True,