summaryrefslogtreecommitdiff
path: root/dtrain/hstreaming/avgweights.rb
blob: d5cfaa4d09bbdd1e8164e92e0d658166f9a16ce2 (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
#!/usr/bin/env ruby1.9.1


STDIN.set_encoding 'utf-8'

#shard_count_key = "__SHARD_COUNT__"

w = {}
c = {}
w.default = 0
c.default = 0
while line = STDIN.gets
  key, val = line.split /\t/
  w[key] += val.to_f
  c[key] += 1.0
end

#shard_count = w["__SHARD_COUNT__"]

w.each_key { |k|
  #if k == shard_count_key then next end
  #if k == "__bias" then next end
  puts "#{k}\t#{w[k]/c[k]}" #{w[k]/shard_count}"
}

#puts "#{shard_count_key}\t#{w[shard_count_key]}"