462a933b3defaf7f10a1e1023b9d81b5e92e3227,ml/rl/models/seq2slate.py,BaselineNet,__init__,#BaselineNet#Any#Any#Any#,353

Before Change


class BaselineNet(nn.Module):
    def __init__(self, state_dim, dim_feedforward, num_stacked_layers):
        super(BaselineNet, self).__init__()
        h_sizes = [state_dim] + [dim_feedforward] * num_stacked_layers + [1]
        self.num_stacked_layers = num_stacked_layers
        self.hidden = nn.ModuleList()
        for k in range(len(h_sizes) - 1):
            self.hidden.append(nn.Linear(h_sizes[k], h_sizes[k + 1]))

    def forward(self, input: rlt.PreprocessedRankingInput):
        x = input.state.float_features
        for i in range(self.num_stacked_layers + 1):
            if i == self.num_stacked_layers:

After Change


class BaselineNet(nn.Module):
    def __init__(self, state_dim, dim_feedforward, num_stacked_layers):
        super(BaselineNet, self).__init__()
        nn_blocks = [nn.Linear(state_dim, dim_feedforward), nn.ReLU()]
        assert num_stacked_layers >= 1
        for _ in range(num_stacked_layers - 1):
            nn_blocks.extend([nn.Linear(dim_feedforward, dim_feedforward), nn.ReLU()])
        nn_blocks.append(nn.Linear(dim_feedforward, 1))
Italian Trulli
In pattern: SUPERPATTERN

Frequency: 3

Non-data size: 6

Instances


Project Name: facebookresearch/Horizon
Commit Name: 462a933b3defaf7f10a1e1023b9d81b5e92e3227
Time: 2020-02-29
Author: czxttkl@fb.com
File Name: ml/rl/models/seq2slate.py
Class Name: BaselineNet
Method Name: __init__


Project Name: chainer/chainerrl
Commit Name: 06bade5c5fc40c0241ae5960b6a0dd1766e7732b
Time: 2017-02-28
Author: muupan@gmail.com
File Name: chainerrl/q_functions/state_action_q_functions.py
Class Name: FCSAQFunction
Method Name: __init__


Project Name: chainer/chainerrl
Commit Name: 939512c36bf89a9ddb02b01818018e4fbfb9bef1
Time: 2017-02-22
Author: muupan@gmail.com
File Name: chainerrl/q_functions/state_action_q_functions.py
Class Name: FCSAQFunction
Method Name: __init__