Merge pull request #17 from conzett/master

Allow passing through file extensions to graph
This commit is contained in:
Ilya Grigorik
2014-02-03 12:23:04 -08:00

View File

@@ -131,10 +131,10 @@ module DecisionTree
descend(@tree, test) descend(@tree, test)
end end
def graph(filename) def graph(filename, file_type = "png")
require 'graphr' require 'graphr'
dgp = DotGraphPrinter.new(build_tree) dgp = DotGraphPrinter.new(build_tree)
dgp.write_to_file("#{filename}.png", "png") dgp.write_to_file("#{filename}.#{file_type}", file_type)
end end
def ruleset def ruleset