toy.py 文件源码

def save_image_fake(fake_data, filename):
    #import warnings
    #warnings.filterwarnings("ignore", category=FutureWarning)
    #import numpy as np
    #import matplotlib
    #matplotlib.use('Agg')
    #import matplotlib.pyplot as plt

    fig, ax = plt.subplots()
    #plt.scatter(real_data[:,0], real_data[:,1], color='blue', label='real')
    plt.scatter(fake_data[:,0], fake_data[:,1], color='red', label='fake')
    plt.axis('equal')
    #plt.legend(loc='upper right', fancybox=True, shadow=True, fontsize=11)
    plt.grid(True)
    plt.xlim(-25, 25)
    plt.ylim(-25, 25)
    plt.minorticks_on()
    plt.xlabel('x', fontsize=14, color='black')
    plt.ylabel('y', fontsize=14, color='black')
    #plt.title('Toy dataset')
    plt.savefig(filename)
    plt.close()