tabensemb.model.AbstractNN.cal_backward_step#

method

AbstractNN.cal_backward_step(loss)[source]#

Perform the backward propagation and optimization steps.

Parameters:

loss: The loss returned by loss_fn().

Notes

Other attributes recorded in loss_fn() can be also used.