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.

# 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