def __init__()

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

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

    def __init__(self, plot_type, fields=None, win=None, env=None, opts={}, port=8097, server="localhost", name=None):
        '''
            Multiple lines can be added to the same plot with the "name" attribute (see example)
            Args:
                fields: Currently unused
                plot_type: {scatter, line}

            Examples:
                >>> scatter_logger = VisdomPlotLogger('line')
                >>> scatter_logger.log(stats['epoch'], loss_meter.value()[0], name="train")
                >>> scatter_logger.log(stats['epoch'], loss_meter.value()[0], name="test")
        '''
        super(VisdomPlotLogger, self).__init__(fields, win, env, opts, port, server)
        valid_plot_types = {
            "scatter": self.viz.scatter,
            "line": self.viz.line}
        self.plot_type = plot_type
        # Set chart type
        if plot_type not in valid_plot_types.keys():
            raise ValueError("plot_type \'{}\' not found. Must be one of {}".format(
                plot_type, valid_plot_types.keys()))
        self.chart = valid_plot_types[plot_type]