lib/xlearn/ffm.rb in xlearn-0.1.0 vs lib/xlearn/ffm.rb in xlearn-0.1.1

@@ -2,7 +2,26 @@
   class FFM < Model
     def initialize(**options)
       @model_type = "ffm"
       super
     end
+
+    # shape is [i, j, k]
+    # for v_{i}_{j}
+    def latent_factors
+      factor = []
+      current = -1
+      read_txt do |line|
+        if line.start_with?("v_")
+          parts = line.split(": ")
+          i = parts.first.split("_")[1].to_i
+          if i != current
+            factor << []
+            current = i
+          end
+          factor.last << parts.last.split(" ").map(&:to_f)
+        end
+      end
+      factor
+    end
   end
 end