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('provola.db', 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     time = Time.now
  24     mkv.learn(line.chomp)
  25     new_ratio = file.pos*100/size
  26     if new_ratio > old_ratio
  27       old_ratio = new_ratio
  28       puts "\n\n\nLearned #{new_ratio}%"
  29       puts "%u words known" % mkv.num_words
  30       min_ord.upto(max_ord) { |ord|
  31         puts "\nOrder #{ord}::"
  32         puts mkv.complete("", :max_order=>ord)
  33       }
  34     end
  35   }
  36 }
  37
  38