This commit is contained in:
bastien
2024-06-25 19:54:35 +02:00
parent 01168f3588
commit 4054395641
2 changed files with 26 additions and 4 deletions

View File

@@ -41,8 +41,8 @@ if "data" in st.session_state:
]
plt.scatter(dataframe_sorted[data_name[0]], dataframe_sorted[target_name], color='b')
plt.scatter(dataframe_sorted[data_name[0]], prediction_array_y, color='r')
else:
plt.plot(dataframe_sorted[data_name[0]], prediction_array_y, color='r')
elif len(data_name) == 2:
ax = fig.add_subplot(111, projection='3d')
prediction_array_y = [
@@ -51,10 +51,9 @@ if "data" in st.session_state:
]
ax.scatter(dataframe_sorted[data_name[0]], dataframe_sorted[data_name[1]], dataframe_sorted[target_name], color='b')
ax.scatter(dataframe_sorted[data_name[0]], dataframe_sorted[data_name[1]], prediction_array_y, color='r')
ax.plot(dataframe_sorted[data_name[0]], dataframe_sorted[data_name[1]], prediction_array_y, color='r')
st.pyplot(fig)
else:
st.error("File not loaded")