Avoid polluting namespace

lib/core_extensions/array.rb

@@ -1,19 +0,0 @@
-class Array
-  def entropy
-    each_with_object(Hash.new(0)) do |i, result|
-      result[i] += 1
-    end.values.inject(0, :+) do |count|
-      percentage = count.to_f / length
-
-      -percentage * Math.log2(percentage)
-    end
-  end
-end
-
-module ArrayClassification
-  refine Array do
-    def classification
-      collect(&:last)
-    end
-  end
-end

lib/core_extensions/object.rb

@@ -1,9 +0,0 @@
-class Object
-  def save_to_file(filename)
-    File.open(filename, "w+") { |f| f << Marshal.dump(self) }
-  end
-
-  def self.load_from_file(filename)
-    Marshal.load(File.read(filename))
-  end
-end

lib/decisiontree.rb

@@ -1,3 +1,4 @@
-require "core_extensions/object"
+require_relative "refinements/object_persistence"
-require "core_extensions/array"
+require_relative "refinements/array_entropy"
-require File.dirname(__FILE__) + "/decisiontree/id3_tree.rb"
+require_relative "refinements/array_classification"
+require_relative "decisiontree/id3_tree"

lib/decisiontree/id3_tree.rb

@@ -7,6 +7,8 @@ module DecisionTree
   Node = Struct.new(:attribute, :threshold, :gain)
 
   using ArrayClassification
+  using ArrayEntropy
+  using ObjectPersistence
 
   class ID3Tree
     def initialize(attributes, data, default, type)

lib/refinements/array_classification.rb

@@ -0,0 +1,7 @@
+module ArrayClassification
+  refine Array do
+    def classification
+      collect(&:last)
+    end
+  end
+end

lib/refinements/array_entropy.rb

@@ -0,0 +1,13 @@
+module ArrayEntropy
+  refine Array do
+    def entropy
+      each_with_object(Hash.new(0)) do |i, result|
+        result[i] += 1
+      end.values.inject(0, :+) do |count|
+        percentage = count.to_f / length
+
+        -percentage * Math.log2(percentage)
+      end
+    end
+  end
+end

lib/refinements/object_persistence.rb

@@ -0,0 +1,11 @@
+module ObjectPersistence
+  refine Object do
+    def save_to_file(filename)
+      File.open(filename, "w+") { |f| f << Marshal.dump(self) }
+    end
+
+    def self.load_from_file(filename)
+      Marshal.load(File.read(filename))
+    end
+  end
+end