fix .drop() axis

autonomio · Apr 20, 2024 · 1cec723 · 1cec723
1 parent c6ac9da
commit 1cec723
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wrangle/df/df_corr_randomforest.py b/wrangle/df/df_corr_randomforest.py
@@ -21,7 +21,7 @@ def df_corr_randomforest(data, y, destructive=True):
     if destructive is False:
         data = data.copy(deep=True)
 
-    x = data.drop(y, 1).values
+    x = data.drop(y, axis=1).values
     labels = data.drop(y, axis=1).columns
     y = data[y].values