git.oblomov.eu Git - rbot-mark/blob - 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.words.length
  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