Source code for svcco.implicit.kernel.kernel_functions.fitness

import numpy as np
import numba as nb

[docs]@nb.jit(nopython=True,fastmath=True) def fitness(x,H_0,dim=3): n = H_0.shape[0]//dim a = np.ones(n*dim) for i in range(n): a[i] = np.cos(x[i*2+1])*np.sin(x[i*2]) a[i+n] = np.sin(x[i*2+1])*np.sin(x[i*2]) a[i+2*n] = np.cos(x[i*2]) return [a.T@H_0@a]