blob: 36d80a197d1ad08502032282991a3f12eecc091b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
module Markov
class Generator
def initialize(inputfile)
@prng = Random.new(Time.now.to_i)
@stats = Marshal::load(File.open(inputfile).read)
@chunklength = @stats["__SETTINGS__"]["__CHUNKLENGTH__"]
end
def cfd(hash)
tot = 0
hash.each_pair {|k, v|
hash[k] = tot += v
}
end
def sel(hash)
max = hash.values.max
r = @prng.rand(max*100000) % max
hash.each_pair {|k, v|
return k if r < v
}
end
def aword(chunk, hash)
sel(cfd(hash[chunk]))
end
def generate(length)
current = @stats.keys.select{|k| k != "__SETTINGS__" }.sample
(length - @chunklength).times do
current << aword(current[-@chunklength..-1], @stats)
end
puts current.join(" ")
end
end
end
|