torchmetrics_validator
TorchmetricsValidator
¶
Bases: BaseValidator
Source code in pytorch_adapt\validators\torchmetrics_validator.py
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
|
__init__(layer='preds', torchmetric_kwargs=None, **kwargs)
¶
Parameters:
Name | Type | Description | Default |
---|---|---|---|
layer |
The name of the layer to pass into the accuracy function.
For example, |
'preds'
|
|
torchmetric_kwargs |
A dictionary of keyword arguments to be passed into the torchmetrics accuracy function. |
None
|
Source code in pytorch_adapt\validators\torchmetrics_validator.py
8 9 10 11 12 13 14 15 16 17 18 19 |
|