import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
import numpy as np

Y_MEM = [37,  3365, 6693, 10021, 13349, 16677, 20005, 23333, 26661]
Y_CPY = [36, 4004, 7972, 11940, 15908,  19876, 23844, 27812, 31780]
YA_MEM = [28, 1788, 3585, 5298, 7028, 9174, 10515, 12258, 14003]
YA_CPY = [31, 2477 , 4909 , 7372, 9802, 12217 , 15049,  17081, 19514]
X = [0, 128, 256, 384, 512, 640, 768, 896, 1024]

plt.scatter(X, Y_MEM, color="b", label="RISCV MEM")
plt.scatter(X, Y_CPY, color="b", marker="x", label="RISCV COPY")
plt.scatter(X, YA_MEM, color="r", label="ARM MEM")
plt.scatter(X, YA_CPY, color="r",marker="x", label="ARM COPY")

plt.xlim([0, 1024])
plt.ylim([50, 32000])
plt.legend()
plt.title("Cycles d'exécution en fonction de n_max")
plt.ylabel("Cycles")
plt.xlabel("N_max")
plt.savefig("memory_cycles.png")
plt.show()