Class: TorchRec::Models::DeepFM::DenseArch

Inherits:
Torch::NN::Module
  • Object
show all
Defined in:
lib/torchrec/models/deepfm/dense_arch.rb

Instance Method Summary collapse

Constructor Details

#initialize(in_features, hidden_layer_size, embedding_dim) ⇒ DenseArch

Returns a new instance of DenseArch.



5
6
7
8
9
10
11
12
13
 
# File 'lib/torchrec/models/deepfm/dense_arch.rb', line 5

def initialize(in_features, hidden_layer_size, embedding_dim)
  super()
  @model = Torch::NN::Sequential.new(
    Torch::NN::Linear.new(in_features, hidden_layer_size),
    Torch::NN::ReLU.new,
    Torch::NN::Linear.new(hidden_layer_size, embedding_dim),
    Torch::NN::ReLU.new
  )
end

Instance Method Details

#forward(features) ⇒ Object 



15
16
17
 
# File 'lib/torchrec/models/deepfm/dense_arch.rb', line 15

def forward(features)
  @model.call(features)
end