Merge pull request #361 from ntumlgroup/fix-sklearn

Eleven1Liu · web-flow · commit 71067046a083 · 2024-02-28T20:29:42.000+08:00
Fixed sklearn version before GridSearchCV is fixed
diff --git a/requirements.txt b/requirements.txt
@@ -2,6 +2,6 @@ liblinear-multicore
 numba
 pandas>1.3.0
 PyYAML
-scikit-learn
+scikit-learn==1.2.2
 scipy
 tqdm
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = libmultilabel
-version = 0.6.0
+version = 0.6.1
 author = LibMultiLabel Team
 license = MIT License
 license_file = LICENSE
@@ -29,7 +29,7 @@ install_requires =
     numba
     pandas>1.3.0
     PyYAML
-    scikit-learn
+    scikit-learn==1.2.2
     scipy
     tqdm
 
