Beispiel #1
0
        print("The dimension is incorrect. Exiting.")
        sys.exit()
    if trval not in results.keys():
        print(trval, " points are not present in results. Exiting.")
        sys.exit()
    for i in range(len(outrange)):

        iout=outrange[i]
        pcf=results['pcmi'][0][iout]
        mindex=results['pcmi'][1][iout]
        #axes_labels=['Model ('+outnames[iout]+')','Polynomial Surrogate']
        print('======== Output # ',str(iout+1),' (QoI ',outnames[iout],') =========')
        print("Plotting model-vs-surrogate at ",trval," points")
        output=results[trval][2][:,iout]
        input=results[trval][0].reshape(-1,ndim)
        fig=io.plot_xy(input[:,d1],output,pnames[d1], outnames[iout],label=trval+' points',savefig='xy_'+pnames[d1]+'_'+outnames[iout]+'.eps')

        ngr=100
        #prange=np.loadtxt('prange.dat',ndmin=2)
        #input_grid_phys=np.linspace(prange[d,0],prange[d,1],ngr)
        input_grid=np.linspace(-1,1,ngr).reshape(-1,1)

        pc_grid=uf.func(input_grid,'PCmi',[mindex,pcf,pctype])

        fig.gca().plot(input_grid,pc_grid,'r-',ms=11,label='Polynomial fit')
        #xdel=prange[d,1]-prange[d,0]
        #gca().set_xlim(prange[d,0]-0.1*xdel,prange[d,1]+0.1*xdel)
        legend()

        fig.savefig('fit_'+pnames[d1]+'_'+outnames[iout]+'_'+trval+'.eps')
        print('Created file fit_'+pnames[d1]+'_'+outnames[iout]+'_'+trval+'.eps')
Beispiel #2
0
    if os.path.exists('outnames.txt'):
        with open('outnames.txt') as f:
            outnames = f.read().splitlines()
            outname = outnames[o]
    else:
        outname = 'Output # ' + str(o)

    print("Plotting %s vs %s" % (pname, outname))

    inputfile = args.inputfile  #sys.argv[4]
    outputfile = args.outputfile  #sys.argv[5]

    input = np.loadtxt(inputfile, ndmin=2)
    output = np.loadtxt(outputfile, ndmin=2)

    io.plot_xy(input[:, d], output[:, o], pname, outname,
               'xy_' + pname + '_' + outname + '.eps')

# Plot output vs input1,input2
elif (plot_type == 'xxy'):

    d1 = int(sys.argv[2])
    d2 = int(sys.argv[3])
    o = int(sys.argv[4])

    if os.path.exists('pnames.txt'):
        with open('pnames.txt') as f:
            pnames = f.read().splitlines()
    else:
        pnames = ['Param # ' + str(d) for d in [d1, d2]]

    if os.path.exists('outnames.txt'):