You should update the class docstring by adding the description for ```loss_type```