Class: TorchRec::Models::DeepFM::OverArch
- Inherits:
-
Torch::NN::Module
- Object
- Torch::NN::Module
- TorchRec::Models::DeepFM::OverArch
- Defined in:
- lib/torchrec/models/deepfm/over_arch.rb
Instance Method Summary collapse
- #forward(features) ⇒ Object
-
#initialize(in_features) ⇒ OverArch
constructor
A new instance of OverArch.
Constructor Details
#initialize(in_features) ⇒ OverArch
Returns a new instance of OverArch.
5 6 7 8 9 10 11 |
# File 'lib/torchrec/models/deepfm/over_arch.rb', line 5 def initialize(in_features) super() @model = Torch::NN::Sequential.new( Torch::NN::Linear.new(in_features, 1), Torch::NN::Sigmoid.new ) end |
Instance Method Details
#forward(features) ⇒ Object
13 14 15 |
# File 'lib/torchrec/models/deepfm/over_arch.rb', line 13 def forward(features) @model.call(features) end |