module XLearn
  class FM < Model
    def initialize(**options)
      @model_type = "fm"
      super
    end

    # shape is [i, k]
    # for v_{i}
    def latent_factors
      factor = []
      read_txt do |line|
        if line.start_with?("v_")
          factor << line.split(": ").last.split(" ").map(&:to_f)
        end
      end
      factor
    end
  end
end