Как использовать пакет simanneal

Question

Я пытаюсь оптимизировать параметры своей функции / объекта, используя имитацию отжига с помощью пакета simanneal https://github.com/perrygeo/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()

Я действительно не знаю, что это значит, хотя. Исходный пример работает со словарем, но моя матрица довольно большая, и я не хочу помещать это в словарь. Кто-нибудь знает, как правильно использовать библиотеку?

Answer 1

Передайте список в это. Сбой из-за оператора if в массиве numpy @ line anneal: 47