summaryrefslogtreecommitdiffstats
path: root/experiments/plot3d.py
blob: b99ef610b8797f799d74c8c65e6ef18e1fa49540 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
import matplotlib.pyplot as plt
import numpy as np

with open("out.log") as fh:
    values = [map(float, line.strip().split()) for line in fh]
    #values = [(b, a, l) for (b, a, l) in values if b >= 0.04]
    am = max(values, key=lambda x: x[4])
    am[0] = 1./am[0]
    print am
    alpha, delta, beta, r , l = zip(*values)
    alpha = 1./np.array(alpha)

    fig = plt.figure(figsize=(12, 8))
    ax = fig.gca(projection='3d')

    ax.plot_trisurf(alpha, delta, l, cmap=cm.jet, linewidth=0.001)
    plt.xlabel("alpha")
    plt.ylabel("delta")
    ax.set_zlabel('Likelihood')
    #plt.savefig("ll.pdf")
    plt.show()