rapidsai / cuml

cuML - RAPIDS Machine Learning Library
https://docs.rapids.ai/api/cuml/stable/
Apache License 2.0
4.25k stars 534 forks source link

[BUG] RFClassifier of regression data, kills kernel - ideally it should error? #5955

Open zard444 opened 4 months ago

zard444 commented 4 months ago

Describe the bug running code kills kernal

Steps/Code to reproduce bug

import cupy
from cuml.datasets import make_regression
from cuml.ensemble import RandomForestClassifier
X, y = make_regression(n_samples=2000, n_features=100, noise=0.1)
model = RandomForestClassifier()
model.fit(X, cupy.ravel(y))

Expected behavior Ideally an error to say its incorrect dtype?

Additional context jupyter lab

dantegd commented 4 months ago

Thanks for the issue @zard444, will try to look into it, not sure what could be the issue, it could be the small memory of the 1650, but I’m not sure of that.