Display amount of known words
[rbot-mark] / test.rb
1 #! /usr/bin/ruby -w
2 # vim: set sw=2 et:
3 # Author: Giuseppe Bilotta <giuseppe.bilotta@gmail.com>
4 # New markov chain plugin tester
5
6 require 'mark2'
7
8 fname = ARGV.fetch(0, "promessi.txt")
9
10 max_ord = ARGV.fetch(1, 5).to_i
11
12 min_ord = ARGV.fetch(2, 5).to_i
13
14 mkv = MarkovChainer.new(max_ord)
15
16 size = File.size?(fname)
17
18 return unless size
19
20 old_ratio = 0
21 File.open(fname) { |file|
22   file.each { |line|
23     mkv.learn(line.chomp)
24     new_ratio = file.pos*100/size
25     if new_ratio > old_ratio
26       old_ratio = new_ratio
27       puts "\n\n\nLearned #{new_ratio}%"
28       puts "%u words known" % mkv.instance_variable_get(:@mkv)[0].size
29       min_ord.upto(max_ord) { |ord|
30         puts "\nOrder #{ord}::"
31         puts mkv.complete("", :max_order=>ord)
32       }
33     end
34   }
35 }
36
37