-
Notifications
You must be signed in to change notification settings - Fork 9
/
base.py
42 lines (30 loc) · 1.36 KB
/
base.py
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# -*- coding: utf-8 -*-
# SPDX-License-Identifier: BSD-3-Clause
from abc import ABC, abstractmethod
class HubnessReduction(ABC):
""" Base class for hubness reduction. """
@abstractmethod
def __init__(self, **kwargs):
pass
@abstractmethod
def fit(self, neigh_dist, neigh_ind, X, assume_sorted, *args, **kwargs):
pass # pragma: no cover
@abstractmethod
def transform(self, neigh_dist, neigh_ind, X, assume_sorted, return_distance=True):
pass # pragma: no cover
def fit_transform(self, neigh_dist, neigh_ind, X, assume_sorted=True, return_distance=True, *args, **kwargs):
""" Equivalent to call .fit().transform() """
self.fit(neigh_dist, neigh_ind, X, assume_sorted, *args, **kwargs)
return self.transform(neigh_dist, neigh_ind, X, assume_sorted, return_distance)
class NoHubnessReduction(HubnessReduction):
""" Compatibility class for neighbor search without hubness reduction. """
def __init__(self, **kwargs):
super().__init__(**kwargs)
def fit(self, *args, **kwargs):
pass # pragma: no cover
def transform(self, neigh_dist, neigh_ind, X, assume_sorted=True, return_distance=True, *args, **kwargs):
""" Equivalent to call .fit().transform() """
if return_distance:
return neigh_dist, neigh_ind
else:
return neigh_ind