Class: XLearn::FM

Inherits:
Model
  • Object
show all
Defined in:
lib/xlearn/fm.rb

Instance Method Summary collapse

Methods inherited from Model

#bias_term, #cv, finalize, finalize_file, #fit, #linear_term, #load_model, #partial_fit, #predict, #save_model, #save_txt

Constructor Details

#initialize(**options) ⇒ FM 



3
4
5
6
 
# File 'lib/xlearn/fm.rb', line 3

def initialize(**options)
  @model_type = "fm"
  super
end

Instance Method Details

#latent_factors ⇒ Object

shape is [i, k] for v_i



10
11
12
13
14
15
16
17
18
 
# File 'lib/xlearn/fm.rb', line 10

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