1
2
3
4
5
6
7
8
9
10
11
12

use pyo3::prelude::*;

use crate::loss_functions::*;

#[pymethods]
impl HuberLoss {
    #[new]
    #[pyo3(signature=(scale=1.0))]
    pub fn new_py(scale: f64) -> Self {
        HuberLoss::new(scale)
    }
}