dkam/decisiontree
1
0
Fork 0
mirror of https://github.com/dkam/decisiontree.git synced 2025-12-28 15:14:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

require graphr only when calling graph(), closes #11

Browse Source
This commit is contained in:
Ilya Grigorik
2013-09-06 19:39:57 -07:00
parent a71572a4f2
commit e70f29817b
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/decisiontree/id3_tree.rb
View File

@@ -3,8 +3,6 @@
### Copyright (c) 2007 Ilya Grigorik <ilya AT igvita DOT com> ### Copyright (c) 2007 Ilya Grigorik <ilya AT igvita DOT com>
### Modifed at 2007 by José Ignacio Fernández <joseignacio.fernandez AT gmail DOT com> ### Modifed at 2007 by José Ignacio Fernández <joseignacio.fernandez AT gmail DOT com>
require 'graphr'
class Object class Object
def save_to_file(filename) def save_to_file(filename)
File.open(filename, 'w+' ) { |f| f << Marshal.dump(self) } File.open(filename, 'w+' ) { |f| f << Marshal.dump(self) }
@@ -133,6 +131,7 @@ module DecisionTree
end end
def graph(filename) def graph(filename)
require 'graphr'
dgp = DotGraphPrinter.new(build_tree) dgp = DotGraphPrinter.new(build_tree)
dgp.write_to_file("#{filename}.png", "png") dgp.write_to_file("#{filename}.png", "png")
end end