b7e09b55-9333-4c49-b273-87a69e5c463f

# coding=utf-8
# Copyright 2018-2023 EvaDB
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import pandas as pd
from thefuzz import fuzz

from evadb.functions.abstract.abstract_function import AbstractFunction


class FuzzDistance(AbstractFunction):
    def setup(self):
        pass

    @property
    def name(self):
        return "FuzzDistance"

    def forward(self, df: pd.DataFrame) -> pd.DataFrame:
        """
        Find the distance between two dataframes using <> distance metric.

        Returns:
            ret (pd.DataFrame): The cropped frame.
        """
        ret = pd.DataFrame()
        ret["distance"] = df.apply(
            lambda row: fuzz.ratio(str(row[0]), str(row[1])), axis=1
        )
        return ret

1	# coding=utf-8
2	# Copyright 2018-2023 EvaDB
3	#
4	# Licensed under the Apache License, Version 2.0 (the "License");
5	# you may not use this file except in compliance with the License.
6	# You may obtain a copy of the License at
7	#
8	# http://www.apache.org/licenses/LICENSE-2.0
9	#
10	# Unless required by applicable law or agreed to in writing, software
11	# distributed under the License is distributed on an "AS IS" BASIS,
12	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13	# See the License for the specific language governing permissions and
14	# limitations under the License.
15	import pandas as pd	×
16	from thefuzz import fuzz	×
17
18	from evadb.functions.abstract.abstract_function import AbstractFunction	×
19
20
21	class FuzzDistance(AbstractFunction):	×
22	def setup(self):	×
23	pass	×
24
25	@property	×
26	def name(self):	×
27	return "FuzzDistance"	×
28
29	def forward(self, df: pd.DataFrame) -> pd.DataFrame:	×
30	"""
31	Find the distance between two dataframes using <> distance metric.
32
33	Returns:
34	ret (pd.DataFrame): The cropped frame.
35	"""
36	ret = pd.DataFrame()	×
37	ret["distance"] = df.apply(	×
38	lambda row: fuzz.ratio(str(row[0]), str(row[1])), axis=1
39	)
40	return ret	×

georgia-tech-db / eva / b7e09b55-9333-4c49-b273-87a69e5c463f

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous