mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-02-20 00:02:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

[mypy] Add/fix type annotations for similarity search in machine learning (#4088)

Browse Source 
* [mypy] Add/fix type annotations for similarity search in machine learning

* fix annotation

* fix annotation (Union)

* isort
This commit is contained in:
Steve Kim 2021-01-22 13:40:21 +09:00 committed by GitHub
parent 7d26ba7075
commit d594f4556c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
machine_learning/similarity_search.py
View File

@ -8,6 +8,7 @@ returns a list containing two data for each vector:
2. distance between the vector and the nearest vector (float)
"""
import math
from typing import List, Union
import numpy as np
@ -30,7 +31,9 @@ def euclidean(input_a: np.ndarray, input_b: np.ndarray) -> float:
return math.sqrt(sum(pow(a - b, 2) for a, b in zip(input_a, input_b)))
def similarity_search(dataset: np.ndarray, value_array: np.ndarray) -> list:
def similarity_search(
dataset: np.ndarray, value_array: np.ndarray
) -> List[List[Union[List[float], float]]]:
"""
:param dataset: Set containing the vectors. Should be ndarray.
:param value_array: vector/vectors we want to know the nearest vector from dataset.