Class: TorchRec::Models::DeepFM::DenseArch
- Inherits:
-
Torch::NN::Module
- Object
- Torch::NN::Module
- TorchRec::Models::DeepFM::DenseArch
- Defined in:
- lib/torchrec/models/deepfm/dense_arch.rb
Instance Method Summary collapse
- #forward(features) ⇒ Object
-
#initialize(in_features, hidden_layer_size, embedding_dim) ⇒ DenseArch
constructor
A new instance of DenseArch.
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, ) 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, ), 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 |