graph-rank 0.0.2 → 0.0.3

data/lib/graph-rank.rb

@@ -1,7 +1,7 @@
 module GraphRank
   
   # Version number.
-  VERSION = '0.0.2'
+  VERSION = '0.0.3'
   
   # Core classes.
   require 'graph-rank/page_rank'

data/lib/graph-rank/page_rank.rb

@@ -4,14 +4,16 @@
 class GraphRank::PageRank
 
   # Initialize with default damping and convergence.
-  def initialize(damping=nil, convergence=nil)
+  # A maximum number of iterations can also be supplied
+  # (default is no maximum, i.e. iterate until convergence).
+  def initialize(damping=nil, convergence=nil, max_it=-1)
     damping ||= 0.85; convergence ||= 0.01
     if damping <= 0 or damping > 1
       raise 'Invalid damping factor.'
     elsif convergence < 0 or convergence > 1
       raise 'Invalid convergence factor.'
     end
-    @damping, @convergence = damping, convergence
+    @damping, @convergence, @max_it = damping, convergence, max_it
     @graph, @outlinks, @nodes, @weights = {}, {}, {}, {}
   end
 
@@ -33,9 +35,11 @@ class GraphRank::PageRank
   def calculate
     done = false
     until done
+      break if @max_it == 0
       new_nodes = iteration
       done = convergence(new_nodes)
       @nodes = new_nodes
+      @max_it -= 1
     end
     @nodes.sort_by {|k,v|v}.reverse
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: graph-rank
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-12-16 00:00:00.000000000 Z
+date: 2012-12-17 00:00:00.000000000 Z
 dependencies: []
 description: ! ' GraphRank is an impementation of TextRank and PageRank in Ruby. '
 email: