Improve code readability

This commit is contained in:
richardpaulhudson 2022-12-09 14:19:19 +01:00
parent abe1cb87c5
commit edc483c379

View File

@ -165,16 +165,14 @@ class EditTreeLemmatizer(TrainablePipe):
eg_d_lowercasing_flags, eg_lowercasing_loss = lowercasing_loss_func( eg_d_lowercasing_flags, eg_lowercasing_loss = lowercasing_loss_func(
eg_lowercasing_flags, self.model.ops.asarray(lowercasing_truths[i]) eg_lowercasing_flags, self.model.ops.asarray(lowercasing_truths[i])
) )
d_scores.append( doc_d_scores = self.model.ops.xp.hstack(
self.model.ops.xp.hstack(
[ [
doc_d_tree_scores, doc_d_tree_scores,
self.model.ops.asarray( self.model.ops.asarray(eg_d_lowercasing_flags, dtype="float32"),
eg_d_lowercasing_flags, dtype="float32"
),
] ]
) )
) d_scores.append(doc_d_scores)
loss += eg_lowercasing_loss loss += eg_lowercasing_loss
else: else:
d_scores, loss = tree_loss_func(scores, tree_truths) d_scores, loss = tree_loss_func(scores, tree_truths)