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** | | | +| **Linear** | | | | **SVM** | | | | **Tree** | | | | **Random Forest** | | | 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(