Merge pull request #5168 from FederatedAI/feature-2.0.0-beta-fate-test

add random state to lr bq example
FederatedAI · Sep 12, 2023 · 5b294e8 · 5b294e8
2 parents e87470d + c7ea9ff
commit 5b294e8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/examples/benchmark_quality/lr/vehicle_config.yaml b/examples/benchmark_quality/lr/vehicle_config.yaml
@@ -6,6 +6,7 @@ epochs: 10
 init_param:
   fit_intercept: True
   method: "random_uniform"
+  random_state: 42
 learning_rate_scheduler:
   method: "linear"
   scheduler_params:

diff --git a/python/requirements-fate.txt b/python/requirements-fate.txt
@@ -7,7 +7,7 @@ pydantic==1.10.12
 cloudpickle==2.1.0
 click
 ruamel-yaml==0.16
-scikit-learn
+scikit-learn==1.0.1
 numpy
 pandas==2.0.3
 transformers==4.30.2

diff --git a/python/setup.py b/python/setup.py
@@ -1,7 +1,9 @@
 import os
-import fate
+
 from setuptools import find_packages, setup
 
+import fate
+
 # Base requirements
 install_requires = [
     "lmdb==1.3.0",
@@ -11,7 +13,7 @@
     "cloudpickle==2.1.0",
     "click",
     "ruamel.yaml==0.16",
-    "scikit-learn",
+    "scikit-learn==1.0.1",
     "numpy",
     "pandas",
     "transformers",