neuromancer.dynamics.library module

class neuromancer.dynamics.library.FourierLibrary(n_features, max_freq=2, include_sin=True, include_cos=True)[source]

Bases: FunctionLibrary

class neuromancer.dynamics.library.FunctionLibrary(functions, n_features, function_names=None)[source]

Bases: object

evaluate(x)[source]
Parameters:

x – (torch.Tensor) an array of values to evaluate the functions at

Returns:

(torch.Tensor, shape=[[# of rows of X, number of functions) the functions evaluated at every single time step of the data

class neuromancer.dynamics.library.PolynomialLibrary(n_features, max_degree=2)[source]

Bases: FunctionLibrary

evaluate(X)[source]
Parameters:

X – (torch.Tensor) the two-dimensional dataset to put through the library

Returns:

(torch.Tensor, [# of rows of X, number of functions]) the library evaluated at X