Method __init__

model_demo/DeepCrossing/layer.py:59–61

(self, hidden_units)

class Res_layer(Layer):
    def __init__(self, hidden_units):
        super(Res_layer, self).__init__()
        self.dense_layer = [MyDense(i, activation="relu") for i in hidden_units]

    def build(self, input_shape):
        self.output_layer = MyDense(input_shape[-1], activation=None)