document exceptions for metrics/regression (#6202)
Co-authored-by: Akihiro Nitta <nitta@akihironitta.com> Co-authored-by: Prajakta Phadke <pphadke@iu.edu> Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
This commit is contained in:
parent
111d9c7267
commit
15c477e9fc
|
@ -59,6 +59,10 @@ class ExplainedVariance(Metric):
|
|||
process_group:
|
||||
Specify the process group on which synchronization is called. default: None (which selects the entire world)
|
||||
|
||||
Raises:
|
||||
ValueError:
|
||||
If ``multioutput`` is not one of ``"raw_values"``, ``"uniform_average"`` or ``"variance_weighted"``.
|
||||
|
||||
Example:
|
||||
|
||||
>>> from pytorch_lightning.metrics import ExplainedVariance
|
||||
|
|
|
@ -51,6 +51,10 @@ class PSNR(Metric):
|
|||
process_group:
|
||||
Specify the process group on which synchronization is called. default: None (which selects the entire world)
|
||||
|
||||
Raises:
|
||||
ValueError:
|
||||
If ``dim`` is not ``None`` and ``data_range`` is not given.
|
||||
|
||||
Example:
|
||||
|
||||
>>> from pytorch_lightning.metrics import PSNR
|
||||
|
|
|
@ -66,6 +66,12 @@ class R2Score(Metric):
|
|||
process_group:
|
||||
Specify the process group on which synchronization is called. default: None (which selects the entire world)
|
||||
|
||||
Raises:
|
||||
ValueError:
|
||||
If ``adjusted`` parameter is not an integer larger or equal to 0.
|
||||
ValueError:
|
||||
If ``multioutput`` is not one of ``"raw_values"``, ``"uniform_average"`` or ``"variance_weighted"``.
|
||||
|
||||
Example:
|
||||
|
||||
>>> from pytorch_lightning.metrics import R2Score
|
||||
|
@ -102,7 +108,7 @@ class R2Score(Metric):
|
|||
self.num_outputs = num_outputs
|
||||
|
||||
if adjusted < 0 or not isinstance(adjusted, int):
|
||||
raise ValueError('`adjusted` parameter should be an integer larger or' ' equal to 0.')
|
||||
raise ValueError('`adjusted` parameter should be an integer larger or equal to 0.')
|
||||
self.adjusted = adjusted
|
||||
|
||||
allowed_multioutput = ('raw_values', 'uniform_average', 'variance_weighted')
|
||||
|
|
Loading…
Reference in New Issue