rarff 0.1.0 → 0.2.0

data/README

@@ -7,18 +7,18 @@ Rarff is a Ruby library for dealing with Attribute-Relation File Format (ARFF) f
 
 == License
 
-Copyright (c) 2006, Andy Payne
+Copyright (c) 2008 Andy Payne
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without
 modification, are permitted provided that the following conditions are met:
 
-	* Redistributions of source code must retain the above copyright notice,
+        * Redistributions of source code must retain the above copyright notice,
         this list of conditions and the following disclaimer.
-	* Redistributions in binary form must reproduce the above copyright notice,
+        * Redistributions in binary form must reproduce the above copyright notice,
         this list of conditions and the following disclaimer in the
         documentation and/or other materials provided with the distribution.
-	* Neither the name of the COPYRIGHT OWNER nor the names of its contributors
+        * Neither the name of the COPYRIGHT OWNER nor the names of its contributors
         may be used to endorse or promote products derived from this software
         without specific prior written permission.
 
@@ -34,6 +34,11 @@ ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 
+== Changes
+
+* Sparse ARFF files (thanks to Tom Adams)
+
+
 == Todo
 
 * Spaces or quotes in nominal types
@@ -63,6 +68,13 @@ Links to documentation:
 * http://www.cs.waikato.ac.nz/~ml/weka/arff.html
 * http://weka.sourceforge.net/wekadoc/index.php/en:ARFF_%283.4.6%29
 
+== Contact Information
+
+Andy Payne
+Website:        http://andy-payne.com/
+Email:          apayne .at. gmail.com
+Twitter:        http://twitter.com/andypayne
+RARFF website:  http://rubyforge.org/projects/rarff/
 
 
 

data/lib/rarff.rb

@@ -199,7 +199,7 @@ class Relation
 	end
 
 
-	def to_arff
+	def to_arff(sparse=false)
 		RELATION_MARKER + " #{@name}\n" +
 		@attributes.map{ |attr| attr.to_arff }.join("\n") +
 		"\n" +
@@ -216,9 +216,13 @@ class Relation
 				elsif @attributes[i].type =~ /^#{ATTRIBUTE_DATE}/i  ## Hack comparison. Ugh.
 					col = '"' + col + '"'
 				end
-				col
-			}.join(', ')
-		}.join("\n")
+				if @attributes[i].type =~ /^#{ATTRIBUTE_NUMERIC}$/i and col == 0
+					nil
+				else
+					sparse ? "#{i} #{col}" : col
+				end
+			}.select{|c|not c.nil?}.join(', ')
+		}.join("\n").gsub(/^/, sparse ? '{' : '').gsub(/$/, sparse ? '}' : '')
 	end
 
 

data/tests/ts_rarff.rb

@@ -5,7 +5,7 @@ require 'rarff'
 
 class TestArffLib < Test::Unit::TestCase
 
-	# Test creation of an arff file string.
+        # Test creation of an arff file string.
     def test_arff_creation
 
     arff_file_str = <<-END_OF_ARFF_FILE
@@ -25,9 +25,9 @@ END_OF_ARFF_FILE
         arff_file_str.gsub!(/\n$/, '')
 
         instances = [ [1.4, 'foo bar', 5, 'baz', "1900-08-08 12:12:12"],
-	        		[20.9, 'ruby', 46, 'rocks', "2005-10-23 12:12:12"],
-		        	[20.9, 'ruby', 46, 'rocks', "2001-02-19 12:12:12"],
-        			[68.1, 'stuff', 728, 'is cool', "1974-02-10 12:12:12"]]
+                                [20.9, 'ruby', 46, 'rocks', "2005-10-23 12:12:12"],
+                                [20.9, 'ruby', 46, 'rocks', "2001-02-19 12:12:12"],
+                                [68.1, 'stuff', 728, 'is cool', "1974-02-10 12:12:12"]]
 
         rel = Rarff::Relation.new('MyCoolRelation')
         rel.instances = instances
@@ -35,38 +35,72 @@ END_OF_ARFF_FILE
         rel.attributes[4].name = 'birthday'
         rel.attributes[4].type = 'DATE "yyyy-MM-dd HH:mm:ss"'
 
-#		puts "rel.to_arff:\n(\n#{rel.to_arff}\n)\n"
+#               puts "rel.to_arff:\n(\n#{rel.to_arff}\n)\n"
         assert_equal(rel.to_arff, arff_file_str, "Arff creation test failed.")
     end
 
+        # Test creation of a sparse arff file string.
+    def test_sparse_arff_creation
 
-	# Test parsing of an arff file.
-	def test_arff_parse
-		in_file = './test_arff.arff'
-		rel = Rarff::Relation.new
-		rel.parse(File.open(in_file).read)
+    arff_file_str = <<-END_OF_ARFF_FILE
+@RELATION MyCoolRelation
+@ATTRIBUTE Attr0 NUMERIC
+@ATTRIBUTE subject STRING
+@ATTRIBUTE Attr2 NUMERIC
+@ATTRIBUTE Attr3 STRING
+@ATTRIBUTE birthday DATE "yyyy-MM-dd HH:mm:ss"
+@DATA
+{0 1.4, 1 'foo bar', 3 baz, 4 "1900-08-08 12:12:12"}
+{0 20.9, 1 ruby, 2 46, 3 rocks, 4 "2005-10-23 12:12:12"}
+{1 ruby, 2 46, 3 rocks, 4 "2001-02-19 12:12:12"}
+{0 68.1, 1 stuff, 3 'is cool', 4 "1974-02-10 12:12:12"}
+END_OF_ARFF_FILE
+
+        arff_file_str.gsub!(/\n$/, '')
+
+        instances = [ [1.4, 'foo bar', 0, 'baz', "1900-08-08 12:12:12"],
+                                [20.9, 'ruby', 46, 'rocks', "2005-10-23 12:12:12"],
+                                [0.0, 'ruby', 46, 'rocks', "2001-02-19 12:12:12"],
+                                [68.1, 'stuff', 0, 'is cool', "1974-02-10 12:12:12"]]
+
+        rel = Rarff::Relation.new('MyCoolRelation')
+        rel.instances = instances
+        rel.attributes[1].name = 'subject'
+        rel.attributes[4].name = 'birthday'
+        rel.attributes[4].type = 'DATE "yyyy-MM-dd HH:mm:ss"'
+
+#               puts "rel.to_arff(true):\n(\n#{rel.to_arff(true)}\n)\n"
+        assert_equal(rel.to_arff(true), arff_file_str, "Arff creation test failed.")
+    end
+
+
+        # Test parsing of an arff file.
+        def test_arff_parse
+                in_file = './test_arff.arff'
+                rel = Rarff::Relation.new
+                rel.parse(File.open(in_file).read)
 
-		assert_equal(rel.instances[2][1], 3.2)
-		assert_equal(rel.instances[7][4], 'Iris-setosa')
-	end
+                assert_equal(rel.instances[2][1], 3.2)
+                assert_equal(rel.instances[7][4], 'Iris-setosa')
+        end
 
 
-	# Test parsing of sparse ARFF format
+        # Test parsing of sparse ARFF format
     def test_sparse_arff_parse
-		in_file = './test_sparse_arff.arff'
-		rel = Rarff::Relation.new
-		rel.parse(File.open(in_file).read)
-
-		assert_equal(rel.instances[0].size, 13)
-		assert_equal(rel.instances[0][1], 0)
-		assert_equal(rel.instances[0][3], 7)
-		assert_equal(rel.instances[1][1], 2.4)
-		assert_equal(rel.instances[1][2], 0)
-		assert_equal(rel.instances[1][12], 19)
-		assert_equal(rel.instances[2][6], 6)
-		assert_equal(rel.instances[3][12], 0)
-#		puts "\n\nARFF: (\n#{rel.to_arff}\n)"
-	end
+                in_file = './test_sparse_arff.arff'
+                rel = Rarff::Relation.new
+                rel.parse(File.open(in_file).read)
+
+                assert_equal(rel.instances[0].size, 13)
+                assert_equal(rel.instances[0][1], 0)
+                assert_equal(rel.instances[0][3], 7)
+                assert_equal(rel.instances[1][1], 2.4)
+                assert_equal(rel.instances[1][2], 0)
+                assert_equal(rel.instances[1][12], 19)
+                assert_equal(rel.instances[2][6], 6)
+                assert_equal(rel.instances[3][12], 0)
+#               puts "\n\nARFF: (\n#{rel.to_arff}\n)"
+        end
 end
 
 

metadata

@@ -1,45 +1,57 @@
 --- !ruby/object:Gem::Specification 
-rubygems_version: 0.8.11
-specification_version: 1
 name: rarff
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
-date: 2006-09-18 00:00:00 -05:00
-summary: Library for handling Weka ARFF files
-require_paths: 
-  - lib
-email: apayne@gmail.com
-homepage: TODO
-rubyforge_project: 
-description: 
-autorequire: rarff
-default_executable: 
-bindir: bin
-has_rdoc: true
-required_ruby_version: !ruby/object:Gem::Version::Requirement 
-  requirements: 
-    - 
-      - ">"
-      - !ruby/object:Gem::Version 
-        version: 0.0.0
-  version: 
+  version: 0.2.0
 platform: ruby
-signing_key: 
-cert_chain: 
 authors: 
-  - Andy Payne
-files: 
-  - lib/rarff.rb
-  - tests/test_arff.arff
-  - tests/test_sparse_arff.arff
-  - tests/ts_rarff.rb
-  - README
-test_files: 
-  - tests/ts_rarff.rb
-rdoc_options: []
-extra_rdoc_files: 
-  - README
+- Andy Payne
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2008-11-22 00:00:00 -06:00
+default_executable: 
+dependencies: []
+
+description: RARFF is a library for handling Weka ARFF files
+email: apayne@gmail.com
 executables: []
+
 extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- README
+- lib/rarff.rb
+- tests/test_arff.arff
+- tests/test_sparse_arff.arff
+- tests/ts_rarff.rb
+has_rdoc: false
+homepage: http://rubyforge.org/projects/rarff/
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
 requirements: []
-dependencies: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: RARFF is a library for handling Weka ARFF files
+test_files: []
+