Spaces:
Sleeping
Sleeping
File size: 1,019 Bytes
a03bfa8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from preprocess.preprocess import Preprocessor
from processor.matching import prepare_groups_with_ids,new_find_matches_with_ids
class Processor():
def __init__(self, long_types_list, short_types_list, sour_list,
type_wine, gbs, colors_for_trim, grapes, other_words,
sour_merge_dict, type_merge_dict, color_merge_dict):
self.preprocessor=Preprocessor(long_types_list, short_types_list, sour_list,
type_wine, gbs, colors_for_trim, grapes, other_words,
sour_merge_dict, type_merge_dict, color_merge_dict)
def process(self, items, products, th=65):
items, products=self.preprocessor.process(self, items, products)
print('-----*-----Matching-----*-----')
items_groups = prepare_groups_with_ids(items)
res=new_find_matches_with_ids(products, items_groups, items, name_threshold=th)
return res.drop(['type','type_wine','year','alco','gb'], axis=1), items, products
|