dask / dask/dask-ml

LinearRegression doesn't return lazy object

Open
#949 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
951
Forks
262
PR merge metrics
No merged PRs in 30d

Description

LinearRegression requires even longer time than sklearn version and it doesn't return a lazy object:

```
%%time

from sklearn.pipeline import make_pipeline
from dask_ml.linear_model import LinearRegression

size = 1e6

X = dask.array.arange(2*size).reshape(-1,2)
y = dask.array.arange(size).reshape(-1,1)
reg = LinearRegression()
reg.fit(X, y)
```
image

It looks as a Dask-incompatible function.

Contributor guide

Open the contributing guide

Research direction

Start with the LinearRegression.fit entry point and reproduce the timing example in the issue using the shown Dask arrays. Compare its behavior with the sklearn version and determine whether fit returns a lazy object. Done means the reported incompatibility and excessive runtime are addressed, with regression coverage added where appropriate.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.