Skip to content

Commit 693ca06

Browse files
committed
check fit_intercept
1 parent 3c9c320 commit 693ca06

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

debug_quantile.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,9 @@ def pinball_loss(residuals, quantile):
1717
X, y = make_regression(n_samples=1000, n_features=10, noise=0.1, random_state=0)
1818
tau = 0.8
1919

20+
fit_intercept = False
2021
start = time.time()
21-
sk = QuantileRegressor(quantile=tau, alpha=0.1, fit_intercept=True)
22+
sk = QuantileRegressor(quantile=tau, alpha=0.1, fit_intercept=fit_intercept)
2223
sk.fit(X, y)
2324
sk_pred = sk.predict(X)
2425
sk_time = time.time() - start
@@ -30,11 +31,11 @@ def pinball_loss(residuals, quantile):
3031
quantile=tau,
3132
alpha=0.1,
3233
delta_init=0.5,
33-
delta_final=0.001,
34+
delta_final=0.00001,
3435
n_deltas=5,
3536
solver="AndersonCD",
3637
verbose=True,
37-
fit_intercept=True
38+
fit_intercept=fit_intercept,
3839
)
3940
qh.fit(X, y)
4041
qh_time = time.time() - start

0 commit comments

Comments
 (0)