2017-04-26 16 views
2

我想通過使用模擬退火通過simanneal包https://github.com/perrygeo/simanneal來優化我的函數/對象的參數。 我的代碼如下:如何使用simanneal包

from simanneal import Annealer 

class ReservoirAnnealer(Annealer): 

    def __init__(self, state, res): 
     self.reservoir = res 
     self.resSize = np.size(self.reservoir.W_top) 
     super(ReservoirAnnealer, self).__init__(state) 


    def move(self): 
     a = random.randint(0, self.resSize) 
     b = random.randint(0, self.resSize) 
     self.state[a,b] ^= 1 

    def energy(self): 
     reservoir.W = self.state 
     return reservoir.fit(data, 1500, penalty=5e-7, errorEvaluationLength=750) 

reservoir = Reservoir(1,1023,spectralRadius=1.25,inputScaling=1,leakingRate=0.3, transientTime=100) 
initialState = reservoir.W_top.ravel() 
annealer = ReservoirAnnealer(initialState, reservoir) 

W_top是0或1裏面,所以像這樣一個矩陣:

[ 1. 1. 1. ..., 0. 0. 1.] 

當我從上面執行的代碼,我得到錯誤:

ValueError        Traceback (most recent call last) 
<ipython-input-87-5a5b9eb8c324> in <module>() 
    20 reservoir = Reservoir(1,1023,spectralRadius=1.25,inputScaling=1,leakingRate=0.3, transientTime=100) 
    21 initialState = reservoir.W_top.ravel() 
---> 22 annealer = ReservoirAnnealer(initialState, reservoir) 
    23 #itinearay, cost = annealer.anneal() 

<ipython-input-87-5a5b9eb8c324> in __init__(self, state, res) 
     6   self.reservoir = res 
     7   #self.resSize = np.size(self.reservoir.W) 
----> 8   super(ReservoirAnnealer, self).__init__(state) 
     9 
    10 

C:\Users\Luca\Anaconda3\lib\site-packages\simanneal\anneal.py in __init__(self, initial_state, load_state) 
    45 
    46  def __init__(self, initial_state=None, load_state=None): 
---> 47   if initial_state: 
    48    self.state = self.copy_state(initial_state) 
    49   elif load_state: 

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all() 

我真的不知道,那是什麼意思。原來的例子與字典一起工作,但我的矩陣非常大,我不想把它放在字典中。 有誰知道如何正確使用庫?

+0

完全回溯? – user3684792

+1

嘗試並傳入一個列表,而不是(init函數 - 這就是爲什麼它無法通過numpy數組上的if語句) – user3684792

+0

噢,謝謝,那工作 –

回答

2

將一個列表傳遞給它。由於numpy陣列上的if語句,線路退火失敗:47