data_prep 0.0.0 → 0.0.1

data/VERSION

@@ -1 +1 @@
-0.0.0
+0.0.1

data/data_prep.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{data_prep}
-  s.version = "0.0.0"
+  s.version = "0.0.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["reddavis"]

data/lib/data_prep.rb

@@ -14,7 +14,7 @@ class DataPrep
   def inject_outputs(data)
     @data = []
     data.each_pair do |output, data|
-      @data << data.map! {|x| [x, [output]]}
+      @data << data.map! {|x| [x, [output].flatten]}
     end
   end
   

data/spec/data_prep_spec.rb

@@ -23,7 +23,18 @@ describe "DataPrep" do
       @data_prep.training[0][0].class.should == Array
       @data_prep.training[0][1].class.should == Array      
     end
+  end
+  
+  describe "When having a target thats array" do
+    before do
+      ham = [[1, 0.5]] * 20
+      spam = [[0, 0.2]] * 20
+      @data_prep = DataPrep.new([0] => ham, [1] => spam)
+    end
     
+    it "should hold 2 arrays" do
+      @data_prep.training[0][1].dimensions.should == 1
+    end
   end
   
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: data_prep
 version: !ruby/object:Gem::Version 
-  version: 0.0.0
+  version: 0.0.1
 platform: ruby
 authors: 
 - reddavis