def get_eth_config()

in visualization/plotting.py [0:0]

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

def get_eth_config():
    nodes = [4, 8, 16, 32]
    fpaths, tags, legends, colors = [], [], [], []
    ar_eth_fpath = 'results_dir/out_files/{tag}out_r{r}_n{n}.csv'
    ar_eth_tags = [
        'AR-DPSGD-CPUCOMM-4ETH-SCRATCH',
        'AR-DPSGD-CPUCOMM-8ETH-SCRATCH',
        'AR-DPSGD-CPUCOMM-16ETH-SCRATCH',
        'AR-DPSGD-CPUCOMM-32ETH-SCRATCH'
    ]
    ar_eth_legends = ['AR-SGD 4 nodes', 'AR-SGD 8 nodes', 'AR-SGD 16 nodes', 'AR-SGD 32 nodes']
    ar_eth_colors = [cm.Reds(x) for x in np.linspace(0.3, 0.8, len(ar_eth_tags))]
    fpaths.append(ar_eth_fpath)
    tags.append(ar_eth_tags)
    legends.append(ar_eth_legends)
    colors.append(ar_eth_colors)

    ds_eth_fpath = 'results_dir/out_files/{tag}out_r{r}_n{n}.csv'
    ds_eth_tags = [
        'SDS-SGD-4ETH',
        'SDS-SGD-8ETH',
        'SDS-SGD-16ETH',
        'SDS-SGD-32ETH'
    ]
    ds_eth_legends = ['D-PSGD 4 nodes', 'D-PSGD 8 nodes', 'D-PSGD 16 nodes', 'D-PSGD 32 nodes']
    ds_eth_colors = [cm.Greens(x) for x in np.linspace(0.3, 0.8, len(ds_eth_tags))]
    fpaths.append(ds_eth_fpath)
    tags.append(ds_eth_tags)
    legends.append(ds_eth_legends)
    colors.append(ds_eth_colors)

    ps_eth_fpath = 'results_dir/out_files/{tag}out_r{r}_n{n}.csv'
    ps_eth_tags = [
        'PS-SGD-4ETH-CHORD',
        'PS-SGD-8ETH-CHORD',
        'PS-SGD-16ETH-CHORD',
        'PS-SGD-32ETH-SCRATCH-CHORD'
    ]
    ps_eth_legends = ['SGP 4 nodes', 'SGP 8 nodes', 'SGP 16 nodes', 'SGP 32 nodes']
    ps_eth_colors = [cm.Blues(x) for x in np.linspace(0.3, 0.8, len(ps_eth_tags))]
    fpaths.append(ps_eth_fpath)
    tags.append(ps_eth_tags)
    legends.append(ps_eth_legends)
    colors.append(ps_eth_colors)

    # return nodes, fpaths[::-1], tags[::-1], legends[::-1], colors[::-1]
    return nodes, fpaths, tags, legends, colors