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)
end
def graph(filename)
def graph(filename, file_type = "png")
require 'graphr'
dgp = DotGraphPrinter.new(build_tree)
dgp.write_to_file("#{filename}.png", "png")
dgp.write_to_file("#{filename}.#{file_type}", file_type)
end
def ruleset