Update FireflyAlgorithm.py

jasonma1127 · Apr 24, 2021 · ece95f1 · ece95f1
1 parent 0062cf5
commit ece95f1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/FireflyAlgorithm.py b/FireflyAlgorithm.py
@@ -1,6 +1,7 @@
 import numpy as np
 import random as rand
 import math
+import time
 
 class FireflyAlgorithm():
 
@@ -40,6 +41,7 @@ def update(self, i, j):
             self.functionArray[i] = self.func(self.tmpArray, self.D)
 
     def doRun(self):
+        start = time.time()
         self.init_FA()
         for gen in range(self.iter_max):
             print("Generation ", gen+1)
@@ -49,4 +51,6 @@ def doRun(self):
                         self.update(i,j)
             print(self.populationArray)
             print(self.functionArray)
+        end = time.time()
+        print("執行時間：%f 秒" % (end - start))
         return self.functionArray.min()