def __init__()

in source/sagemaker/sagemaker_graph_fraud_detection/dgl_fraud_detection/model/pytorch.py [0:0]

• 16 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self,
                 g,
                 in_feats,
                 n_hidden,
                 n_classes,
                 n_layers,
                 activation,
                 dropout):
        super(GCN, self).__init__()
        self.g = g
        self.layers = nn.Sequential()
        # input layer
        self.layers.add_module(GraphConv(in_feats, n_hidden, activation=activation))
        # hidden layers
        for i in range(n_layers - 1):
            self.layers.add(GraphConv(n_hidden, n_hidden, activation=activation))
        # output layer
        # self.layers.add(GraphConv(n_hidden, n_classes))
        self.layers.add(nn.Linear(n_hidden, n_classes))
        self.dropout = nn.Dropout(p=dropout)