LinearRegression doesn't return lazy object
- 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)
```

It looks as a Dask-incompatible function.
Contributor 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