import matplotlib.pyplot as plt
import seaborn as sb
from math import sqrt

sb.set_style("white")

values = [map(float, line.strip().split()) for line in open("results.txt")]
x, y, z = zip(*values)
y = map(sqrt, y)
z = map(sqrt, z)

plt.figure(figsize=(9, 6))
plt.plot(x, y, label="train")
plt.plot(x, z, label="validation")
plt.legend()
plt.xlabel("K")
plt.ylabel("RMSE")
plt.savefig("rmse.pdf")