Closed tji5otma closed 1 year ago
Hi,
You are right, the sklearn wrapper for the Torch version slipped through the unit nan-input tests because of a small mistake on my behalf. This was a simple fix and has been fixed now. Reinstalling PGBM from pip should solve the issue (make sure the version installed is 2.1.1). Please re-open if not fixed.
I am interested in probabilistic GBM and was doing some research and found your PGBM repository. So I read the README and according to the Feature overview, pgbm.torch.PGBM is compatible with NaN. However, when I tried the following code, I got
ValueError: Input X contains NaN.
This is a slight modification of your sample code. If you know anything about this problem, please let me know.