def __init__()

in containers/Shoot/CNN/unet.py [0:0]

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

    def __init__(self, depth=5,activation="relu",drop_out=.1,pool="max",width=5,pad_width=16,**kwargs):
        super(Unet, self).__init__(**kwargs)
        self.config={
            "pad_width":pad_width,
            "depth":depth,
            "activation":activation,
            "drop_out":drop_out,
            "pool":pool,
            "width":width
        }
        self.pad_width=pad_width
        with self.name_scope():
            self.down_branch=Down_Branch(depth=depth,activation=activation,pool=pool,width=width)
            self.up_branch=Up_Branch(depth=depth-1,activation=activation,width=width+3)
            self.activation=nn.Activation("sigmoid")
            self.drop_out=nn.Dropout(drop_out)