Skip to content

Commit d5820e7

Browse files
committed
add cosine similarity kernel (from scikit-learn)
1 parent ead08aa commit d5820e7

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

qstack/regression/kernel_utils.py

+3
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,9 @@ def get_local_kernel(arg):
258258
elif arg=='dot':
259259
from sklearn.metrics.pairwise import linear_kernel
260260
return lambda x,y,s: linear_kernel(x, y)
261+
elif arg=='cosine':
262+
from sklearn.metrics.pairwise import cosine_similarity
263+
return lambda x,y,s: cosine_similarity(x, y)
261264
elif arg=='L':
262265
from sklearn.metrics.pairwise import laplacian_kernel
263266
return laplacian_kernel

0 commit comments

Comments
 (0)