WienerCascadeRegression

class WienerCascadeRegression(degree=3)

Class for the Wiener Cascade Decoder

Parameters:

degree (integer, optional, default 3) – The degree of the polynomial used for the static nonlinearity

fit(X_flat_train, y_train)

Train Wiener Cascade Decoder

Parameters:

• X_flat_train (numpy 2d array of shape [n_samples,n_features]) – This is the neural data. See example file for an example of how to format the neural data correctly

• y_train (numpy 2d array of shape [n_samples, n_outputs]) – This is the outputs that are being predicted

predict(X_flat_test)

Predict outcomes using trained Wiener Cascade Decoder

Parameters:

X_flat_test (numpy 2d array of shape [n_samples,n_features]) – This is the neural data being used to predict outputs.

Returns:

y_test_predicted – The predicted outputs

Return type:

numpy 2d array of shape [n_samples,n_outputs]