explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-03-03 10:55:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add get_cosine util function

Browse Source
This commit is contained in:
Ines Montani 2017-01-12 16:49:57 +01:00
parent 51ef75f629
commit bd20ec0a6a
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/tests/util.py
View File

@ -4,6 +4,8 @@ from __future__ import unicode_literals
from ..tokens import Doc from ..tokens import Doc
from ..attrs import ORTH, POS, HEAD, DEP from ..attrs import ORTH, POS, HEAD, DEP
import numpy
def get_doc(vocab, words=[], pos=None, heads=None, deps=None, tags=None, ents=None): def get_doc(vocab, words=[], pos=None, heads=None, deps=None, tags=None, ents=None):
"""Create Doc object from given vocab, words and annotations.""" """Create Doc object from given vocab, words and annotations."""
@ -36,3 +38,8 @@ def apply_transition_sequence(parser, doc, sequence):
with parser.step_through(doc) as stepwise: with parser.step_through(doc) as stepwise:
for transition in sequence: for transition in sequence:
stepwise.transition(transition) stepwise.transition(transition)
def get_cosine(vec1, vec2):
"""Get cosine for two given vectors"""
return numpy.dot(vec1, vec2) / (numpy.linalg.norm(vec1) * numpy.linalg.norm(vec2))