FYS3150/Project-4
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Cory Balaton
2023-12-03 16:42:50 +01:00
parent ef1d54b776
commit 83468938bb
5 changed files with 63 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
python_scripts/pd_estimate.py
View File

@@ -1,4 +1,4 @@
import os
from os import makedirs
from pathlib import Path
import matplotlib.pyplot as plt
@@ -9,20 +9,20 @@ from scipy.stats import norm
sns.set_theme()
params = {
"font.family": "Serif",
"font.serif": "Roman",
"font.serif": "Roman",
"text.usetex": True,
"axes.titlesize": "large",
"axes.labelsize": "large",
"xtick.labelsize": "large",
"ytick.labelsize": "large",
"legend.fontsize": "medium"
"legend.fontsize": "medium",
}
plt.rcParams.update(params)
def plot(infile, outfile):
if not (outdir := Path(outfile).parent).exists():
os.makedirs(outdir)
makedirs(outdir)
figure1, ax1 = plt.subplots()
eps = []
@@ -32,21 +32,32 @@ def plot(infile, outfile):
vals = line.strip().split(",")
eps.append(float(vals[0]))
ax1.hist(eps, np.arange(min(eps), max(eps) + 0.02, 0.02), color="steelblue", density=True, ec="white", lw=0.2)
ax1.hist(
eps,
np.arange(min(eps), max(eps) + 0.02, 0.02),
color="steelblue",
density=True,
ec="white",
lw=0.2,
)
ax1.set_xlabel(r"$\epsilon$")
ax1.set_ylabel("Density")
mu, sigma = np.mean(eps), np.std(eps)
mu, sigma = np.mean(eps), np.std(eps)
x = np.arange(min(eps), max(eps) + 0.02, 0.02)
stats = (f"$\\langle \\epsilon \\rangle$ = {np.mean(eps):.4f}\n"
f"Var$(\\epsilon)$ = {np.var(eps):.4f}")
stats = (
f"$\\langle \\epsilon \\rangle$ = {np.mean(eps):.4f}\n"
f"Var$(\\epsilon)$ = {np.var(eps):.4f}"
)
bbox = dict(boxstyle="round", pad=0.3, fc="white", ec="white", alpha=0.5)
ax1.text(0.95, 0.9, stats, bbox=bbox, transform=ax1.transAxes, ha="right", va="center")
ax1.text(
0.95, 0.9, stats, bbox=bbox, transform=ax1.transAxes, ha="right", va="center"
)
ax1.plot(x, norm.pdf(x, mu, sigma), color="mediumseagreen")
figure1.savefig(outfile, bbox_inches="tight")
if __name__ == "__main__":
plot("output/pd_estimate/estimate_1_0.txt", "../latex/images/pd_estimate_1_0.pdf")
plot("output/pd_estimate/estimate_2_4.txt", "../latex/images/pd_estimate_2_4.pdf")
plot("./data/hp/pd_estimate/estimate_1_0.txt", "./latex/images/pd_estimate_1_0.pdf")
plot("./data/hp/pd_estimate/estimate_2_4.txt", "./latex/images/pd_estimate_2_4.pdf")