diff --git a/felimination/rfe.py b/felimination/rfe.py index 12e36f3..7228cdf 100644 --- a/felimination/rfe.py +++ b/felimination/rfe.py @@ -67,8 +67,8 @@ def _train_score_get_importance( X_train, y_train = _safe_split(estimator, X, y, train) X_test, y_test = _safe_split(estimator, X, y, test, train) estimator = estimator.fit(X_train, y_train) - train_score = _score(estimator, X_train, y_train, scorer) - test_score = _score(estimator, X_test, y_test, scorer) + train_score = _score(estimator, X_train, y_train, scorer, score_params=None) + test_score = _score(estimator, X_test, y_test, scorer, score_params=None) importances = _get_feature_importances( estimator, importance_getter, X=X_test, y=y_test ) diff --git a/pyproject.toml b/pyproject.toml index 6f6925b..761cbc0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "felimination" description = "This library contains some useful scikit-learn compatible classes for feature selection." -version = "0.2.2" +version = "0.2.3" keywords = ["feature selection", "scikit-learn", "machine learning"] authors = [ { name = "Claudio Salvatore Arcidiacono", email = "author@email.com" } @@ -15,7 +15,7 @@ classifiers = [ ] dependencies = [ - "scikit-learn>=1.0.1,<2.0.0", + "scikit-learn>=1.4.0,<2.0.0", "pandas>=1.1.1,<3.0.0", "seaborn>=0.11.1,<1.0.0", ] diff --git a/requirements.txt b/requirements.txt index 0aa4515..f8d1e0f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,40 +1,40 @@ -black==23.3.0 -cfgv==3.3.1 -click==8.1.3 -contourpy==1.1.0 -cycler==0.11.0 -distlib==0.3.6 -filelock==3.12.2 -flake8==6.0.0 +black==24.4.2 +cfgv==3.4.0 +click==8.1.7 +contourpy==1.2.1 +cycler==0.12.1 +distlib==0.3.8 +filelock==3.14.0 +flake8==7.0.0 Flake8-pyproject==1.2.3 -fonttools==4.40.0 -identify==2.5.24 +fonttools==4.52.4 +identify==2.5.36 iniconfig==2.0.0 -joblib==1.3.1 -kiwisolver==1.4.4 -matplotlib==3.7.1 +joblib==1.4.2 +kiwisolver==1.4.5 +matplotlib==3.9.0 mccabe==0.7.0 mypy-extensions==1.0.0 nodeenv==1.8.0 -numpy==1.25.0 -packaging==23.1 -pandas==2.0.3 -pathspec==0.11.1 -Pillow==10.0.0 -platformdirs==3.8.0 -pluggy==1.2.0 -pre-commit==3.3.3 -pycodestyle==2.10.0 -pyflakes==3.0.1 -pyparsing==3.1.0 -pytest==7.4.0 -python-dateutil==2.8.2 -pytz==2023.3 -PyYAML==6.0 -scikit-learn==1.3.0 -scipy==1.11.1 -seaborn==0.12.2 +numpy==1.26.4 +packaging==24.0 +pandas==2.2.2 +pathspec==0.12.1 +pillow==10.3.0 +platformdirs==4.2.2 +pluggy==1.5.0 +pre-commit==3.7.1 +pycodestyle==2.11.1 +pyflakes==3.2.0 +pyparsing==3.1.2 +pytest==8.2.1 +python-dateutil==2.9.0.post0 +pytz==2024.1 +PyYAML==6.0.1 +scikit-learn==1.5.0 +scipy==1.13.1 +seaborn==0.13.2 six==1.16.0 -threadpoolctl==3.1.0 -tzdata==2023.3 -virtualenv==20.23.1 +threadpoolctl==3.5.0 +tzdata==2024.1 +virtualenv==20.26.2