def __init__()

in src/model.py [0:0]

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

    def __init__(self,
                 decoder,
                 image_encoder,
                 maxnumlabels,
                 crit=None,
                 crit_eos=None,
                 crit_cardinality=None,
                 pad_value=0,
                 perminv=True,
                 decoder_ff=False,
                 th=0.5,
                 loss_label='bce',
                 replacement=False,
                 card_type='none',
                 dataset='voc',
                 U=2.36,
                 use_empty_set=False,
                 eps=1e-8):

        super(SetPred, self).__init__()
        self.image_encoder = image_encoder
        self.decoder = decoder
        self.decoder_ff = decoder_ff
        self.maxnumlabels = maxnumlabels
        self.crit = crit
        self.th = th
        self.perminv = perminv
        self.pad_value = pad_value
        self.crit_eos = crit_eos
        self.crit_cardinality = crit_cardinality
        self.loss_label = loss_label
        self.replacement = replacement
        self.card_type = card_type
        self.dataset = dataset
        self.u_term = math.log(U)
        self.eps = eps
        self.use_empty_set = use_empty_set