Thanks to visit codestin.com
Credit goes to github.com

Skip to content

ENH Euclidean specialization of DatasetsPair instead of ArgKmin and RadiusNeighbors #25170

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 10 commits into from
Closed

ENH Euclidean specialization of DatasetsPair instead of ArgKmin and RadiusNeighbors #25170

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
0322b27
first commit
Vincent-Maladiere Dec 12, 2022
cd5db58
fix csr csr case
Vincent-Maladiere Dec 16, 2022
d615c04
fix csr csr 32bits
Vincent-Maladiere Dec 17, 2022
52d751e
fix argkmin, radius_neighbors still fails most tests
Vincent-Maladiere Dec 18, 2022
63bba0c
simplify argkmin to try to improve on asv benchmarking, but segfault …
Vincent-Maladiere Dec 18, 2022
9fbeace
move dist_middle_terms_chunks from middle_term_computer to datasets_pair
Vincent-Maladiere Dec 22, 2022
95cedcc
further optim tentatives for EuclideanDenseDenseDatasetsPair
Vincent-Maladiere Dec 23, 2022
73a4535
Merge branch 'main' into middle_term_computer_revamp
jjerphan Jan 4, 2023
59be06d
inline, finalize and remove default thread_num
Vincent-Maladiere Jan 6, 2023
32e8b8a
fixup! inline, finalize and remove default thread_num
Vincent-Maladiere Jan 6, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 0 additions & 11 deletions sklearn/metrics/_pairwise_distances_reduction/_argkmin.pxd.tp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@ cnp.import_array()
{{for name_suffix in ['64', '32']}}

from ._base cimport BaseDistancesReduction{{name_suffix}}
from ._middle_term_computer cimport MiddleTermComputer{{name_suffix}}

cdef class ArgKmin{{name_suffix}}(BaseDistancesReduction{{name_suffix}}):
"""float{{name_suffix}} implementation of the ArgKmin."""
Expand All @@ -21,14 +20,4 @@ cdef class ArgKmin{{name_suffix}}(BaseDistancesReduction{{name_suffix}}):
DTYPE_t ** heaps_r_distances_chunks
ITYPE_t ** heaps_indices_chunks


cdef class EuclideanArgKmin{{name_suffix}}(ArgKmin{{name_suffix}}):
"""EuclideanDistance-specialisation of ArgKmin{{name_suffix}}."""
cdef:
MiddleTermComputer{{name_suffix}} middle_term_computer
const DTYPE_t[::1] X_norm_squared
const DTYPE_t[::1] Y_norm_squared

bint use_squared_distances

{{endfor}}
Loading