Skip to content

Commit

Permalink
Replaced LinearRegressor with DummyRegressor
Browse files Browse the repository at this point in the history
  • Loading branch information
SiddhantSadangi committed Jan 16, 2024
1 parent 4e9d2cd commit bb1b212
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tests/test_e2e.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import pytest
from sklearn import datasets
from sklearn.cluster import KMeans
from sklearn.dummy import DummyRegressor
from sklearn.linear_model import (
LinearRegression,
LogisticRegression,
Expand Down Expand Up @@ -75,11 +76,11 @@ def test_unsupported_object():
X, y = datasets.load_diabetes(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.5)

model = LinearRegression()
model = DummyRegressor()

param_grid = {
"copy_X": [True, False],
"fit_intercept": [True, False],
"strategy": ["mean", "median", "quantile"],
"quantile": [0.1, 0.5, 1.0],
}

X, y = datasets.fetch_california_housing(return_X_y=True)[:10]
Expand Down

0 comments on commit bb1b212

Please sign in to comment.