Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

Commit bb7f35a

Browse filesBrowse files
committed
MAINT Adapt DistanceMetric for fused sparse-dense
To use the a more efficient implementation for future callers implementations. See: #15
1 parent 5fd66bc commit bb7f35a
Copy full SHA for bb7f35a

File tree

3 files changed

+230
-288
lines changed
Filter options

3 files changed

+230
-288
lines changed

‎sklearn/metrics/_dist_metrics.pxd.tp

Copy file name to clipboardExpand all lines: sklearn/metrics/_dist_metrics.pxd.tp
+7-7Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -101,9 +101,9 @@ cdef class DistanceMetric{{name_suffix}}:
101101

102102
cdef DTYPE_t dist_csr(
103103
self,
104-
const {{INPUT_DTYPE_t}}[:] x1_data,
104+
const {{INPUT_DTYPE_t}}* x1_data,
105105
const SPARSE_INDEX_TYPE_t[:] x1_indices,
106-
const {{INPUT_DTYPE_t}}[:] x2_data,
106+
const {{INPUT_DTYPE_t}}* x2_data,
107107
const SPARSE_INDEX_TYPE_t[:] x2_indices,
108108
const SPARSE_INDEX_TYPE_t x1_start,
109109
const SPARSE_INDEX_TYPE_t x1_end,
@@ -114,9 +114,9 @@ cdef class DistanceMetric{{name_suffix}}:
114114

115115
cdef DTYPE_t rdist_csr(
116116
self,
117-
const {{INPUT_DTYPE_t}}[:] x1_data,
117+
const {{INPUT_DTYPE_t}}* x1_data,
118118
const SPARSE_INDEX_TYPE_t[:] x1_indices,
119-
const {{INPUT_DTYPE_t}}[:] x2_data,
119+
const {{INPUT_DTYPE_t}}* x2_data,
120120
const SPARSE_INDEX_TYPE_t[:] x2_indices,
121121
const SPARSE_INDEX_TYPE_t x1_start,
122122
const SPARSE_INDEX_TYPE_t x1_end,
@@ -140,7 +140,7 @@ cdef class DistanceMetric{{name_suffix}}:
140140

141141
cdef int pdist_csr(
142142
self,
143-
const {{INPUT_DTYPE_t}}[:] x1_data,
143+
const {{INPUT_DTYPE_t}}* x1_data,
144144
const SPARSE_INDEX_TYPE_t[:] x1_indices,
145145
const SPARSE_INDEX_TYPE_t[:] x1_indptr,
146146
const ITYPE_t size,
@@ -149,10 +149,10 @@ cdef class DistanceMetric{{name_suffix}}:
149149

150150
cdef int cdist_csr(
151151
self,
152-
const {{INPUT_DTYPE_t}}[:] x1_data,
152+
const {{INPUT_DTYPE_t}}* x1_data,
153153
const SPARSE_INDEX_TYPE_t[:] x1_indices,
154154
const SPARSE_INDEX_TYPE_t[:] x1_indptr,
155-
const {{INPUT_DTYPE_t}}[:] x2_data,
155+
const {{INPUT_DTYPE_t}}* x2_data,
156156
const SPARSE_INDEX_TYPE_t[:] x2_indices,
157157
const SPARSE_INDEX_TYPE_t[:] x2_indptr,
158158
const ITYPE_t size,

0 commit comments

Comments
0 (0)
Morty Proxy This is a proxified and sanitized view of the page, visit original site.