better integration, chart.to_json

1 files changed, 12 insertions, 7 deletions

diff --git a/test_hg.rb b/test_hg.rb
index f4e2ef8..14fe011 100755
--- a/test_hg.rb
+++ b/test_hg.rb
@@ -3,17 +3,22 @@
 require_relative 'hg'
 
 
+# viterbi
 semiring = ViterbiSemiring.new
-hypergraph, nodes_by_id = HG::read_hypergraph_from_json('example/3/3.json', semiring, true)
+hypergraph, nodes_by_id = HG::read_hypergraph_from_json('example/toy/toy.json', semiring, true)
+#hypergraph, nodes_by_id = HG::read_hypergraph_from_json('example/glue/glue.json', semiring, true)
+#hypergraph, nodes_by_id = HG::read_hypergraph_from_json('example/3/3.json', semiring, true)
 path, score = HG::viterbi_path hypergraph, nodes_by_id[-1], semiring
 s = HG::derive path, path.last.head, []
 puts "#{s.map { |i| i.word }.join ' '}"
 puts Math.log score
 puts
-#hypergraph.reset
-#paths = HG::all_paths hypergraph, nodes_by_id[-1]
-#paths.each_with_index { |p,i|
-#  s = HG::derive p, p.last.head, []
-#  puts "#{i+1}. #{s.map { |i| i.word }.join ' '}"
-#}
+
+# all paths
+hypergraph.reset
+paths = HG::all_paths hypergraph, nodes_by_id[-1]
+paths.each_with_index { |p,i|
+  s = HG::derive p, p.last.head, []
+  puts "#{i+1}. #{s.map { |x| x.word }.join ' '}"
+}