def __addlogger()

in torchnet/logger/meterlogger.py [0:0]

• 23 lines of code
• 4 McCabe index (conditional complexity)

    def __addlogger(self, meter, ptype):
        if ptype == 'line':
            if self.plotstylecombined:
                opts = {'title': self.title + ' ' + meter}
                self.logger['Train'][meter] = VisdomPlotLogger(ptype, env=self.env, server=self.server,
                                                               port=self.port, opts=opts)
                opts = {}
                self.logger['Test'][meter] = self.logger['Train'][meter]
            else:
                opts = {'title': self.title + 'Train ' + meter}
                self.logger['Train'][meter] = VisdomPlotLogger(ptype, env=self.env, server=self.server,
                                                               port=self.port, opts=opts)
                opts = {'title': self.title + 'Test ' + meter}
                self.logger['Test'][meter] = VisdomPlotLogger(ptype, env=self.env, server=self.server,
                                                              port=self.port, opts=opts)
        elif ptype == 'heatmap':
            names = list(range(self.nclass))
            opts = {'title': self.title + ' Train ' + meter, 'columnnames': names, 'rownames': names}
            self.logger['Train'][meter] = VisdomLogger('heatmap', env=self.env, server=self.server,
                                                       port=self.port, opts=opts)
            opts = {'title': self.title + ' Test ' + meter, 'columnnames': names, 'rownames': names}
            self.logger['Test'][meter] = VisdomLogger('heatmap', env=self.env, server=self.server,
                                                      port=self.port, opts=opts)