Replies: 2 comments 1 reply
-
Thank you for reporting this. |
Beta Was this translation helpful? Give feedback.
-
Thank you for your response.
Below is the code I'm running that I run into the value error mentioned above.
|
Beta Was this translation helpful? Give feedback.
-
I am splitting the data with set_sample_splitting. I only get this error when I manually sample split and don't get it with the same data when I don't set sample split - wonder if this function is not working.
smpls = (train_idx, test_idx) # Correctly formatted tuple
Set the sample splitting for the DoubleMLPLR object
dml_plr_obj.set_sample_splitting(smpls)
dml_plr_obj.fit()
but I keep on getting this below error after fitting:
/usr/local/lib/python3.10/dist-packages/sklearn/utils/validation.py in _assert_all_finite_element_wise(X, xp, allow_nan, msg_dtype, estimator_name, input_name)
173 "#estimators-that-handle-nan-values"
174 )
--> 175 raise ValueError(msg_err)
176
177
ValueError: Input contains NaN.
Beta Was this translation helpful? Give feedback.
All reactions