brainpy.nn.runners.BPFF
brainpy.nn.runners.BPFF#
- class brainpy.nn.runners.BPFF(target, **kwargs)[source]#
The trainer implementing back propagation algorithm for feedforward neural networks.
Methods
__init__
(target, **kwargs)f_grad
([shared_kwargs])Get gradient function.
f_loss
([shared_kwargs])Get loss function.
f_train
([shared_kwargs])Get training function.
fit
(train_data[, test_data, num_batch, ...])Fit the target model according to the given training and testing data.
predict
(xs[, initial_states, ...])Predict a series of input data with the given target model.
Attributes
mapping_type
Mapping type for the output and the target.
train_losses
Training loss.