@@ -4917,7 +4917,10 @@ <h2 id="spotpython.utils.repair.apply_penalty_NA" class="doc doc-heading">
49174917
49184918 < details class ="quote ">
49194919 < summary > Source code in < code > spotpython/utils/repair.py</ code > </ summary >
4920- < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 94</ span >
4920+ < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 91</ span >
4921+ < span class ="normal "> 92</ span >
4922+ < span class ="normal "> 93</ span >
4923+ < span class ="normal "> 94</ span >
49214924< span class ="normal "> 95</ span >
49224925< span class ="normal "> 96</ span >
49234926< span class ="normal "> 97</ span >
@@ -4966,10 +4969,7 @@ <h2 id="spotpython.utils.repair.apply_penalty_NA" class="doc doc-heading">
49664969< span class ="normal "> 140</ span >
49674970< span class ="normal "> 141</ span >
49684971< span class ="normal "> 142</ span >
4969- < span class ="normal "> 143</ span >
4970- < span class ="normal "> 144</ span >
4971- < span class ="normal "> 145</ span >
4972- < span class ="normal "> 146</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> apply_penalty_NA</ span > < span class ="p "> (</ span > < span class ="n "> y</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> penalty_NA</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="p "> ,</ span > < span class ="n "> sd</ span > < span class ="o "> =</ span > < span class ="mf "> 0.1</ span > < span class ="p "> ,</ span > < span class ="n "> stop_on_zero_return</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> ,</ span > < span class ="n "> verbosity</ span > < span class ="o "> =</ span > < span class ="mi "> 0</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> :</ span >
4972+ < span class ="normal "> 143</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> apply_penalty_NA</ span > < span class ="p "> (</ span > < span class ="n "> y</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> penalty_NA</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="p "> ,</ span > < span class ="n "> sd</ span > < span class ="o "> =</ span > < span class ="mf "> 0.1</ span > < span class ="p "> ,</ span > < span class ="n "> stop_on_zero_return</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> ,</ span > < span class ="n "> verbosity</ span > < span class ="o "> =</ span > < span class ="mi "> 0</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> :</ span >
49734973< span class ="w "> </ span > < span class ="sd "> """</ span >
49744974< span class ="sd "> Replaces NaN values in y with a penalty value of penalty_NA and issues a warning if necessary.</ span >
49754975
@@ -5202,10 +5202,7 @@ <h2 id="spotpython.utils.repair.remove_nan" class="doc doc-heading">
52025202< span class ="normal "> 85</ span >
52035203< span class ="normal "> 86</ span >
52045204< span class ="normal "> 87</ span >
5205- < span class ="normal "> 88</ span >
5206- < span class ="normal "> 89</ span >
5207- < span class ="normal "> 90</ span >
5208- < span class ="normal "> 91</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> remove_nan</ span > < span class ="p "> (</ span > < span class ="n "> X</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> y</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> stop_on_zero_return</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tuple</ span > < span class ="p "> [</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ]:</ span >
5205+ < span class ="normal "> 88</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> remove_nan</ span > < span class ="p "> (</ span > < span class ="n "> X</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> y</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> stop_on_zero_return</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tuple</ span > < span class ="p "> [</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ]:</ span >
52095206< span class ="w "> </ span > < span class ="sd "> """Remove rows from X and y where y contains NaN values and issue a warning</ span >
52105207< span class ="sd "> if the dimension of the returned y array is smaller than the dimension of the original y array.</ span >
52115208< span class ="sd "> Handles both 1D (shape `(n,)`) and 2D (shape `(n, m)`) y arrays.</ span >
@@ -5238,8 +5235,6 @@ <h2 id="spotpython.utils.repair.remove_nan" class="doc doc-heading">
52385235< span class ="sd "> >>> print(X_cleaned, y_cleaned)</ span >
52395236< span class ="sd "> [[1 2]] [[1. 2.]]</ span >
52405237< span class ="sd "> """</ span >
5241- < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="sa "> f</ span > < span class ="s2 "> "remove_nan: X.shape=</ span > < span class ="si "> {</ span > < span class ="n "> X</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="si "> }</ span > < span class ="s2 "> , y.shape=</ span > < span class ="si "> {</ span > < span class ="n "> y</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="si "> }</ span > < span class ="s2 "> "</ span > < span class ="p "> )</ span >
5242- < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="sa "> f</ span > < span class ="s2 "> "remove_nan: X=</ span > < span class ="si "> {</ span > < span class ="n "> X</ span > < span class ="si "> }</ span > < span class ="s2 "> ,</ span > < span class ="se "> \n</ span > < span class ="s2 "> y=</ span > < span class ="si "> {</ span > < span class ="n "> y</ span > < span class ="si "> }</ span > < span class ="s2 "> "</ span > < span class ="p "> )</ span >
52435238 < span class ="c1 "> # Get the original dimension of the y array</ span >
52445239 < span class ="n "> original_dim</ span > < span class ="o "> =</ span > < span class ="n "> y</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="p "> [</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span >
52455240
@@ -5250,7 +5245,6 @@ <h2 id="spotpython.utils.repair.remove_nan" class="doc doc-heading">
52505245 < span class ="n "> ind</ span > < span class ="o "> =</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> all</ span > < span class ="p "> (</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> isfinite</ span > < span class ="p "> (</ span > < span class ="n "> y</ span > < span class ="p "> ),</ span > < span class ="n "> axis</ span > < span class ="o "> =</ span > < span class ="mi "> 0</ span > < span class ="p "> )</ span >
52515246 < span class ="k "> else</ span > < span class ="p "> :</ span >
52525247 < span class ="k "> raise</ span > < span class ="ne "> ValueError</ span > < span class ="p "> (</ span > < span class ="s2 "> "y must be a 1D or 2D array."</ span > < span class ="p "> )</ span >
5253- < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="sa "> f</ span > < span class ="s2 "> "remove_nan: ind=</ span > < span class ="si "> {</ span > < span class ="n "> ind</ span > < span class ="si "> }</ span > < span class ="s2 "> "</ span > < span class ="p "> )</ span >
52545248
52555249 < span class ="c1 "> # Update X and y by removing rows with NaN in y</ span >
52565250 < span class ="n "> X_cleaned</ span > < span class ="o "> =</ span > < span class ="n "> X</ span > < span class ="p "> [</ span > < span class ="n "> ind</ span > < span class ="p "> ,</ span > < span class ="p "> :]</ span >
0 commit comments