Class: TextRank::RankFilter::NormalizeProbability

Inherits:
Object
  • Object
show all
Defined in:
lib/text_rank/rank_filter/normalize_probability.rb

Overview

A rank filter which normalizes the ranked keywords so that the sum of the rank values is 1.0 (a "probability" normalization).

= Example

NormalizeProbability.new.filter!( { "town" => 0.6818754334834477, "cities" => 0.6055017128817066, "siege" => 0.5411519524982207, "arts" => 0.4907977453782612, "envy" => 0.4692709808107252, "blessings" => 0.4442147897516214, "plagues" => 0.3972420789430091, "florish" => 0.2746092797528525, "devoured" => 0.26867321734332237, "anxieties" => 0.2367731719604189, "peace" => 0.1905352582752693, "inhabitants" => 0.02715120116732137, } ) => { "town" => 0.1473434248897056, "cities" => 0.13084016782478722, "siege" => 0.11693511476062682, "arts" => 0.10605429845557579, "envy" => 0.10140267579486278, "blessings" => 0.09598839508602595, "plagues" => 0.08583827125543537, "florish" => 0.0593390959673909, "devoured" => 0.058056398684529435, "anxieties" => 0.051163259981992296, "peace" => 0.041171915188530236, "inhabitants" => 0.005866982110537665, }

Instance Method Summary collapse

Instance Method Details

#filter!(ranks, **_) ⇒ Hash<String, Float>

Perform the filter on the ranks

Parameters:

  • ranks (Hash<String, Float>) —

    the results of the PageRank algorithm

Returns:

  • (Hash<String, Float>) 


45
46
47
48
49
50
 
# File 'lib/text_rank/rank_filter/normalize_probability.rb', line 45

def filter!(ranks, **_)
  return if ranks.empty?

  total = ranks.values.reduce(:+)
  ranks.transform_values { |v| v / total }
end