Function plot_with_domains

project_utils/visualization_utils.py:95–161

(save_path, lowDWeights, labels)

def plot_with_domains(save_path, lowDWeights, labels):
    import matplotlib.pyplot as plt
    padding_rate = 0.1
    plt.cla()
    X, Y = lowDWeights[:, 0], lowDWeights[:, 1]
    unique = np.unique(labels)

    num_type_color = 7
    colors = sns.color_palette("viridis", n_colors=num_type_color)
    # label2color_dict = {l: cmap(i) for i, l in enumerate(unique)}
    box_alpha = 0.3
    for x, y, s in zip(X, Y, labels):
        if s >= 0 and s < 100:
            plt.text(x, y, s, fontsize=1,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle', alpha=box_alpha,  # 圆圈
                               ec=colors[0],  # 边框颜色
                               fc=colors[0]  # 填充颜色
                               ))
        elif s >= 100 and s < 200:
            plt.text(x, y, s, fontsize=1,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle', alpha=box_alpha,  # 圆圈
                               ec=colors[1],  # 边框颜色
                               fc=colors[1]  # 填充颜色
                               ))
        elif s >= 200 and s < 300:
            plt.text(x, y, s, fontsize=1,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle', alpha=box_alpha,  # 圆圈
                               ec=colors[2],  # 边框颜色
                               fc=colors[2]  # 填充颜色
                               ))
        elif s >= 300 and s < 400:
            plt.text(x, y, s, fontsize=1,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle', alpha=box_alpha,  # 圆圈
                               ec=colors[3],  # 边框颜色
                               fc=colors[3]  # 填充颜色
                               ))
        elif s >= 400 and s < 500:
            plt.text(x, y, s, fontsize=1,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle', alpha=box_alpha,  # 圆圈
                               ec=colors[4],  # 边框颜色
                               fc=colors[4]  # 填充颜色
                               ))
        else:
            plt.text(x, y, s, fontsize=5,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle', alpha=box_alpha,  # 圆圈
                               ec=colors[5],  # 边框颜色
                               fc=colors[5]  # 填充颜色
                               ))
    # plt.xlim(X.min()-padding_rate*X.min(), X.max()+padding_rate*X.max())
    # plt.ylim(Y.min()-padding_rate*Y.min(), Y.max()+padding_rate*Y.max())
    # plt.legend()
    plt.xlim(X.min(), X.max())