ruby-svd 0.4.0 → 0.4.5

data/lib/lsa.rb

@@ -11,16 +11,18 @@ class LSA
   # and all the clusters (columns) used in the original matrix.
   # Returns a sorted list of indexes and distances,
   def classify_vector(values)
-    raise "Unsupported vector length" unless values.size == @u.row_size
+    raise "Unsupported vector length" unless values.size == @u.row_size || values.size == @v.row_size
     vector = Matrix.row_vector(values)
-    position = vector * @u * @s.inverse
-    puts position
+    mult_matrix = (values.size == @u.row_size ? @u : @v)
+    comp_matrix = (values.size == @u.row_size ? @v : @u)
+    
+    position = vector * mult_matrix * @s.inverse
     x = position[0,0]
     y = position[0,1]
     results = []
     
-    @v.row_size.times do |index|
-      results << [index, cosine_similarity(x, y, @v[index, 0], @v[index, 1])]
+    comp_matrix.row_size.times do |index|
+      results << [index, cosine_similarity(x, y, comp_matrix[index, 0], comp_matrix[index, 1])]
     end
     
     results.sort {|a, b| b[1] <=> a[1]}

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 4
-  - 0
-  version: 0.4.0
+  - 5
+  version: 0.4.5
 platform: ruby
 authors: 
 - Will Cannings
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-03-18 00:00:00 +11:00
+date: 2010-03-19 00:00:00 +11:00
 default_executable: 
 dependencies: []