diff options
Diffstat (limited to 'convert_to_svmlight_format')
-rwxr-xr-x | convert_to_svmlight_format | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/convert_to_svmlight_format b/convert_to_svmlight_format new file mode 100755 index 0000000..7e5c538 --- /dev/null +++ b/convert_to_svmlight_format @@ -0,0 +1,21 @@ +#!/usr/bin/env ruby + +require 'zipf' + +fd = Marshal.load ReadFile.read ARGV[0] +d = fd.size + +train = [] +l_i = 1 +while line = STDIN.gets + puts l_i if l_i%1000==0 + s = [] + line.split.each { |i| + k,w = i.split '=', 2 + s << [fd[k]+1, w.to_f] + } + s.sort_by! { |i| i.first } + puts "+1 #{s.map{|i| "#{i.first}:#{i[1]}" }.join(' ')}" + l_i+= 1 +end + |