fix typos

12 months ago · b5feac89cb
parent 1de7f3c512
commit b5feac89cb
2 changed files with 1 additions and 2 deletions
--- a/src/back/managing_missing_values.py
+++ b/src/back/managing_missing_values.py
@ -48,7 +48,7 @@ def impute_with_regression(data):
        model = LinearRegression()
        model.fit(X_complete, y_complete)
        y_pred = model.predict(X_missing)
-        data.loc[df[col].isnull(), col] = y_pred
+        data.loc[data[col].isnull(), col] = y_pred
        
    return data

--- a/src/pages/clean.py
+++ b/src/pages/clean.py
@ -13,7 +13,6 @@ if 'original_df' in st.session_state:
    st.write("## Missing data")
    rm_empty_rows_or_cols = st.checkbox("Remove empty rows or columns", True)

-
    st.write("#### Replace missing values")
    replace_methods = ["mean","median","mode","knn","regression"]
    replace_method = st.radio('Choose an option:', replace_methods)