Update tdcs_problem.py

ncclabsustech · Nov 1, 2023 · 537a6da · 537a6da
1 parent c96119e
commit 537a6da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tdcs_problem.py b/tdcs_problem.py
@@ -2,7 +2,7 @@
 from public import glo
 import geatpy as ea
 from public import util
-
+NUM_ELE = glo.NUM_ELE
 
 class MyProblem(ea.Problem): 
 
@@ -14,7 +14,7 @@ def __init__(self):
         Dim = 4  
         varTypes = [1] * Dim  
         lb = [0, 0,0,0]  
-        ub = [74, 74,74,74]  
+        ub = [(NUM_ELE-1), (NUM_ELE-1),(NUM_ELE-1),(NUM_ELE-1)]  
         lbin = [1] * Dim  
         ubin = [1] * Dim