diff --git a/README.md b/README.md
index 76f1400e..26ef945c 100644
--- a/README.md
+++ b/README.md
@@ -42,7 +42,7 @@ pip install m2cgen
| | Classification | Regression |
| --- | --- | --- |
-| **Linear** |
- scikit-learn
- LogisticRegression
- LogisticRegressionCV
- PassiveAggressiveClassifier
- Perceptron
- RidgeClassifier
- RidgeClassifierCV
- SGDClassifier
- lightning
- AdaGradClassifier
- CDClassifier
- FistaClassifier
- SAGAClassifier
- SAGClassifier
- SDCAClassifier
- SGDClassifier
| - scikit-learn
- ARDRegression
- BayesianRidge
- ElasticNet
- ElasticNetCV
- HuberRegressor
- Lars
- LarsCV
- Lasso
- LassoCV
- LassoLars
- LassoLarsCV
- LassoLarsIC
- LinearRegression
- OrthogonalMatchingPursuit
- OrthogonalMatchingPursuitCV
- PassiveAggressiveRegressor
- RANSACRegressor(only supported regression estimators can be used as a base estimator)
- Ridge
- RidgeCV
- SGDRegressor
- TheilSenRegressor
- StatsModels
- Generalized Least Squares (GLS)
- Generalized Least Squares with AR Errors (GLSAR)
- Ordinary Least Squares (OLS)
- Weighted Least Squares (WLS)
- lightning
- AdaGradRegressor
- CDRegressor
- FistaRegressor
- SAGARegressor
- SAGRegressor
- SDCARegressor
|
+| **Linear** | - scikit-learn
- LogisticRegression
- LogisticRegressionCV
- PassiveAggressiveClassifier
- Perceptron
- RidgeClassifier
- RidgeClassifierCV
- SGDClassifier
- lightning
- AdaGradClassifier
- CDClassifier
- FistaClassifier
- SAGAClassifier
- SAGClassifier
- SDCAClassifier
- SGDClassifier
| - scikit-learn
- ARDRegression
- BayesianRidge
- ElasticNet
- ElasticNetCV
- HuberRegressor
- Lars
- LarsCV
- Lasso
- LassoCV
- LassoLars
- LassoLarsCV
- LassoLarsIC
- LinearRegression
- OrthogonalMatchingPursuit
- OrthogonalMatchingPursuitCV
- PassiveAggressiveRegressor
- RANSACRegressor(only supported regression estimators can be used as a base estimator)
- Ridge
- RidgeCV
- SGDRegressor
- TheilSenRegressor
- StatsModels
- Generalized Least Squares (GLS)
- Generalized Least Squares with AR Errors (GLSAR)
- Ordinary Least Squares (OLS)
- Quantile Regression (QuantReg)
- Weighted Least Squares (WLS)
- lightning
- AdaGradRegressor
- CDRegressor
- FistaRegressor
- SAGARegressor
- SAGRegressor
- SDCARegressor
|
| **SVM** | | |
| **Tree** | - DecisionTreeClassifier
- ExtraTreeClassifier
| - DecisionTreeRegressor
- ExtraTreeRegressor
|
| **Random Forest** | - ExtraTreesClassifier
- LGBMClassifier(rf booster only)
- RandomForestClassifier
- XGBRFClassifier(binary only, multiclass is not supported yet)
| - ExtraTreesRegressor
- LGBMRegressor(rf booster only)
- RandomForestRegressor
- XGBRFRegressor
|
diff --git a/tests/e2e/test_e2e.py b/tests/e2e/test_e2e.py
index 120382a2..f2b4fe18 100644
--- a/tests/e2e/test_e2e.py
+++ b/tests/e2e/test_e2e.py
@@ -264,6 +264,9 @@ def classification_binary_random(model):
regression(utils.StatsmodelsSklearnLikeWrapper(
sm.OLS,
dict(fit_regularized=STATSMODELS_LINEAR_REGULARIZED_PARAMS))),
+ regression(utils.StatsmodelsSklearnLikeWrapper(
+ sm.QuantReg,
+ dict(init=dict(fit_intercept=True)))),
regression(utils.StatsmodelsSklearnLikeWrapper(
sm.WLS,
dict(init=dict(fit_intercept=True, weights=np.arange(