pub enum LossKind {
MSE,
BCE,
}
Expand description
Supported losses
Variants§
MSE
Mean Squared Error: L = 1/m * sum((y_hat - y)^2)
BCE
Binary Cross-Entropy: L = -1/m * sum(y*log(y_hat) + (1-y)*log(1-y_hat))
Implementations§
Trait Implementations§
Auto Trait Implementations§
impl Freeze for LossKind
impl RefUnwindSafe for LossKind
impl Send for LossKind
impl Sync for LossKind
impl Unpin for LossKind
impl UnwindSafe for LossKind
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more