Skip to content

Commit

Permalink
Document supported versions of packages (#197)
Browse files Browse the repository at this point in the history
  • Loading branch information
StrikerRUS authored Apr 27, 2020
1 parent 70f6bef commit f1caa20
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 4 deletions.
3 changes: 3 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,9 @@ pip install m2cgen
| **Random Forest** | <ul><li>ExtraTreesClassifier</li><li>LGBMClassifier(rf booster only)</li><li>RandomForestClassifier</li><li>XGBRFClassifier(binary only, multiclass is not supported yet)</li></ul> | <ul><li>ExtraTreesRegressor</li><li>LGBMRegressor(rf booster only)</li><li>RandomForestRegressor</li><li>XGBRFRegressor</li></ul> |
| **Boosting** | <ul><li>LGBMClassifier(gbdt/dart/goss booster only)</li><li>XGBClassifier(gbtree/gblinear booster only)</li><ul> | <ul><li>LGBMRegressor(gbdt/dart/goss booster only)</li><li>XGBRegressor(gbtree/gblinear booster only)</li></ul> |

You can find versions of packages with which compatibility is guaranteed by CI tests [here](https://github.com/BayesWitnesses/m2cgen/blob/master/requirements-test.txt#L1).
Other versions can also be supported but they are untested.

## Classification Output
### Linear / Linear SVM / Kernel SVM
#### Binary
Expand Down
13 changes: 9 additions & 4 deletions requirements-test.txt
Original file line number Diff line number Diff line change
@@ -1,13 +1,18 @@
numpy==1.18.2
scipy==1.4.1
# Supported models
scikit-learn==0.22.2.post1
xgboost==1.0.2
lightgbm==2.3.1
statsmodels==0.11.1
git+git://github.com/scikit-learn-contrib/lightning.git@b96f9c674968496e854078163c8814049a7b9f43

# Testing tools
flake8==3.7.9
pytest==5.4.1
pytest-mock==3.0.0
coveralls==2.0.0
pytest-cov==2.8.1

# Other stuff
numpy==1.18.2
scipy==1.4.1
py-mini-racer==0.2.0
statsmodels==0.11.1
git+git://github.com/scikit-learn-contrib/lightning.git@b96f9c674968496e854078163c8814049a7b9f43

0 comments on commit f1caa20

Please sign in to comment.