iut/miner
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

update

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

23
frontend/pages/prediction_classification.py
View File

@@ -4,6 +4,8 @@ from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score from sklearn.metrics import accuracy_score
from sklearn.preprocessing import LabelEncoder from sklearn.preprocessing import LabelEncoder
import pandas as pd import pandas as pd
import matplotlib.pyplot as plt
st.header("Prediction: Classification") st.header("Prediction: Classification")
@@ -60,5 +62,26 @@ if "data" in st.session_state:
prediction = label_encoders[target_name].inverse_transform(prediction) prediction = label_encoders[target_name].inverse_transform(prediction)
st.write("Prediction:", prediction[0]) st.write("Prediction:", prediction[0])
fig = plt.figure()
dataframe_sorted = pd.concat([X, y], axis=1).sort_values(by=data_name)
X = dataframe_sorted[data_name[0]]
y = dataframe_sorted[target_name]
prediction_array_y = [
model.predict(pd.DataFrame([[dataframe_sorted[data_name[0]].iloc[i]]], columns=data_name))[0]
for i in range(dataframe_sorted.shape[0])
]
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')
st.pyplot(fig)
else: else:
st.error("File not loaded") st.error("File not loaded")

7
frontend/pages/prediction_regression.py
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]], dataframe_sorted[target_name], color='b')
plt.scatter(dataframe_sorted[data_name[0]], prediction_array_y, color='r') plt.plot(dataframe_sorted[data_name[0]], prediction_array_y, color='r')
else: elif len(data_name) == 2:
ax = fig.add_subplot(111, projection='3d') ax = fig.add_subplot(111, projection='3d')
prediction_array_y = [ 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]], 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) st.pyplot(fig)
else: else:
st.error("File not loaded") st.error("File not loaded")