Fixed bug with power iteration

miso-belica · Aug 11, 2023 · 8d29125 · 8d29125
1 parent 2f6c959
commit 8d29125
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sumy/summarizers/lex_rank.py b/sumy/summarizers/lex_rank.py
@@ -164,6 +164,7 @@ def power_method(matrix, epsilon):
 
  while lambda_val > epsilon:
  next_p = numpy.dot(transposed_matrix, p_vector)
+ next_p /= numpy.linalg.norm(next_p)
  lambda_val = numpy.linalg.norm(numpy.subtract(next_p, p_vector))
  p_vector = next_p