Sklearn random forest setting random state

chemprop · Sep 24, 2020 · 69c29ed · 69c29ed
1 parent a387d18
commit 69c29ed
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/chemprop/sklearn_train.py b/chemprop/sklearn_train.py
@@ -227,7 +227,7 @@ def run_sklearn(args: SklearnTrainArgs,
     debug('Building model')
     if args.dataset_type == 'regression':
         if args.model_type == 'random_forest':
-            model = RandomForestRegressor(n_estimators=args.num_trees, n_jobs=-1)
+            model = RandomForestRegressor(n_estimators=args.num_trees, n_jobs=-1, random_state=args.seed)
         elif args.model_type == 'svm':
             model = SVR()
         else:

diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -182,12 +182,12 @@ def interpret(self,
         (
                 'sklearn_random_forest',
                 'random_forest',
-                1.575406
+                1.582733
         ),
         (
                 'sklearn_svm',
                 'svm',
-                1.698937
+                1.698927
         ),
         (
                 'chemprop',
@@ -278,7 +278,7 @@ def test_train_multi_task_classification(self,
         (
                 'sklearn_random_forest',
                 'random_forest',
-                0.915988
+                0.945589
         ),
         (
                 'sklearn_svm',