From 5d2de4d0ec008b3e5737ad8b2ee531a9dc88e1c3 Mon Sep 17 00:00:00 2001
From: duiyipan <azula_fire@163.com>
Date: Sat, 14 Sep 2024 11:37:08 +0800
Subject: [PATCH] add random seed

---
 expo/experimenter/autosklearn.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/expo/experimenter/autosklearn.py b/expo/experimenter/autosklearn.py
index 7340edafa..22aa2a132 100644
--- a/expo/experimenter/autosklearn.py
+++ b/expo/experimenter/autosklearn.py
@@ -17,6 +17,7 @@ def create_autosklearn_scorer(metric_name):
 
 class ASRunner:
     time_limit = 300
+    seed = 42
 
     def __init__(self, state=None):
         self.state = state
@@ -49,6 +50,7 @@ class ASRunner:
                 per_run_time_limit=60,
                 metric=create_autosklearn_scorer(eval_metric),
                 memory_limit=8192,
+                seed=self.seed,
                 tmp_folder="AutosklearnModels/as-{}-{}".format(
                     self.state["task"], datetime.now().strftime("%y%m%d_%H%M")
                 ),
@@ -60,6 +62,7 @@ class ASRunner:
                 per_run_time_limit=60,
                 metric=create_autosklearn_scorer(eval_metric),
                 memory_limit=8192,
+                seed=self.seed,
                 tmp_folder="AutosklearnModels/as-{}-{}".format(
                     self.state["task"], datetime.now().strftime("%y%m%d_%H%M")
                 ),