Adding experiments code

1 files changed, 21 insertions, 0 deletions

diff --git a/experiments/plot.py b/experiments/plot.py
new file mode 100644
index 0000000..cbea485
--- /dev/null
+++ b/experiments/plot.py
@@ -0,0 +1,21 @@
+from mpl_toolkits.mplot3d import Axes3D
+from matplotlib import cm
+import matplotlib.pyplot as plt
+import numpy as np
+
+with open("results3.txt") 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[2])
+    print am
+    beta, alpha, l = zip(*values)
+
+    fig = plt.figure(figsize=(12, 8))
+    ax = fig.gca(projection='3d')
+
+    ax.plot_trisurf(beta, alpha, l, cmap=cm.jet, linewidth=0.001)
+    plt.xlabel("Beta")
+    plt.ylabel("Alpha")
+    ax.set_zlabel('Likelihood')
+    #plt.savefig("ll.pdf")
+    plt.show()