Apply suggestions from code review

Co-authored-by: Sofie Van Landeghem <svlandeg@users.noreply.github.com>
This commit is contained in:
Daniël de Kok 2023-01-13 16:22:19 +01:00 committed by GitHub
parent 8d725d9d9c
commit 985dd2236c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 4 deletions

View File

@ -70,7 +70,7 @@ cdef class TrainablePipe(Pipe):
teacher_pipe (Optional[TrainablePipe]): The teacher pipe to learn teacher_pipe (Optional[TrainablePipe]): The teacher pipe to learn
from. from.
examples (Iterable[Example]): Distillation examples. The eference examples (Iterable[Example]): Distillation examples. The reference
and predicted docs must have the same number of tokens and the and predicted docs must have the same number of tokens and the
same orthography. same orthography.
drop (float): dropout rate. drop (float): dropout rate.

View File

@ -221,7 +221,7 @@ cdef class Parser(TrainablePipe):
teacher_pipe (Optional[TrainablePipe]): The teacher pipe to learn teacher_pipe (Optional[TrainablePipe]): The teacher pipe to learn
from. from.
examples (Iterable[Example]): Distillation examples. The eference examples (Iterable[Example]): Distillation examples. The reference
and predicted docs must have the same number of tokens and the and predicted docs must have the same number of tokens and the
same orthography. same orthography.
drop (float): dropout rate. drop (float): dropout rate.
@ -315,8 +315,6 @@ cdef class Parser(TrainablePipe):
RETURNS (Tuple[float, float]): The loss and the gradient. RETURNS (Tuple[float, float]): The loss and the gradient.
RETURNS (Tuple[float, float]): The loss and the gradient.
DOCS: https://spacy.io/api/dependencyparser#get_teacher_student_loss DOCS: https://spacy.io/api/dependencyparser#get_teacher_student_loss
""" """
loss_func = LegacySequenceCategoricalCrossentropy(normalize=False) loss_func = LegacySequenceCategoricalCrossentropy(normalize=False)