Update supervised.py to resolve SettingWithCopyWarning

sarahshi · May 28, 2024 · 90fb446 · 90fb446
1 parent 75f4d3b
commit 90fb446
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/mineralML/supervised.py b/src/mineralML/supervised.py
@@ -167,7 +167,7 @@ def norm_data_nn(df):
     # scaled_df = df[oxides].reset_index(drop=True).copy()
 
     if df[oxides].isnull().any().any():
-        df, _ = prep_df_nn(df)
+        df = prep_df_nn(df)
     else:
         df = df
 
@@ -528,10 +528,10 @@ def predict_class_prob_nn(df, n_iterations=250):
     first_predict_mineral = class2mineral_nn(top_two_indices[:, 1])
     second_predict_mineral = class2mineral_nn(top_two_indices[:, 0])
 
-    df["Predict_Mineral"] = first_predict_mineral
-    df["Predict_Probability"] = first_predict_prob
-    df["Second_Predict_Mineral"] = second_predict_mineral
-    df["Second_Predict_Probability"] = second_predict_prob
+    df.loc[:, "Predict_Mineral"] = first_predict_mineral
+    df.loc[:, "Predict_Probability"] = first_predict_prob
+    df.loc[:, "Second_Predict_Mineral"] = second_predict_mineral
+    df.loc[:, "Second_Predict_Probability"] = second_predict_prob
 
     return df, probability_matrix