Spaces:
Sleeping
Sleeping
File size: 879 Bytes
bd39f54 10c7c36 bd39f54 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import numpy as np
import matplotlib.pyplot as plt
from static.config import Config
# draw scatter line graph
def draw_scatter_line_graph(x_data, y_pred_data, y_real_data, coef, intercept, labels, title):
# Manually adjust based on the data
layout = """
ABCDE
FGHIJ
"""
fig, ax = plt.subplot_mosaic(layout, figsize=(16, 16))
for i in range(np.size(x_data, 1)):
ax[str(chr(i+65))].scatter(x_data[:, i], y_pred_data.T, color=Config.COLORS[0], s=4, label=labels[0])
ax[str(chr(i+65))].scatter(x_data[:, i], y_real_data, color=Config.COLORS[1], s=4, label=labels[1])
ax[str(chr(i+65))].plot(x_data[:, i], x_data[:, i] * coef[i] + intercept, color=Config.COLORS[2], markersize=4)
ax[str(chr(i + 65))].legend()
plt.suptitle(title)
plt.savefig("./diagram/{}.png".format(title), dpi=300)
plt.show()
|