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

add r2 score

Browse Source
This commit is contained in:
bastien ollier
2024-06-26 11:05:04 +02:00
parent da1e97f07f
commit 9bc9e21e45
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
frontend/pages/prediction_regression.py
View File

@@ -22,10 +22,6 @@ if "data" in st.session_state:
model = LinearRegression()
model.fit(X, y)
y_pred = [model.predict(pd.DataFrame([pred_value[0]], columns=data_name)) for pred_value in X.values.tolist()]
r2 = r2_score(y, y_pred)
st.write('R-squared score:', r2)
st.subheader("Enter values for prediction")
pred_values = [st.number_input(f"Value for {feature}", value=0.0) for feature in data_name]
prediction = model.predict(pd.DataFrame([pred_values], columns=data_name))
@@ -35,8 +31,11 @@ if "data" in st.session_state:
fig = plt.figure()
dataframe_sorted = pd.concat([X, y], axis=1).sort_values(by=data_name)
if len(data_name) == 1:
y_pred = [model.predict(pd.DataFrame([pred_value[0]], columns=data_name)) for pred_value in X.values.tolist()]
r2 = r2_score(y, y_pred)
st.write('R-squared score:', r2)
X = dataframe_sorted[data_name[0]]
y = dataframe_sorted[target_name]