a  d @sddlZddlmZmZddlZddlmZddlmZddlm Z ddl m Z m Z m Z eeddd Zdeeeed eeeefed ddZeedddZdeeed eeeefedddZdeed eeeefedddZdS)N)OptionalUnion)Tensor)Literal)#_multiclass_confusion_matrix_update)_drop_empty_rows_and_cols_handle_nan_in_data_nominal_input_validation)confmatreturncCsVt|}|}||}|d|}|dd|jd}t|t||S)aCompute Conditional Entropy Statistic based on a pre-computed confusion matrix. .. math:: H(X|Y) = \sum_{x, y ~ (X, Y)} p(x, y)\frac{p(y)}{p(x, y)} Args: confmat: Confusion matrix for observed data Returns: Conditional Entropy Value )rsumZ unsqueezerepeatshapetorchZnansumlog)r total_occurrencesZp_xy_mZp_yZp_y_mrq/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchmetrics/functional/nominal/theils_u.py_conditional_entropy_computes rreplace)rZdrop)predstarget num_classes nan_strategynan_replace_valuer cCsN|jdkr|dn|}|jdkr,|dn|}t||||\}}t|||S)aCompute the bins to update the confusion matrix with for Theil's U calculation. Args: preds: 1D or 2D tensor of categorical (nominal) data target: 1D or 2D tensor of categorical (nominal) data num_classes: Integer specifing the number of classes nan_strategy: Indication of whether to replace or drop ``NaN`` values nan_replace_value: Value to replace ``NaN`s when ``nan_strategy = 'replace``` Returns: Non-reduced confusion matrix r )ndimZargmaxrr)rrrrrrrr_theils_u_update7srcCs`t|}t|}|}|d|}t|t| }|dkrTtjd|jdS|||S)zCompute Theil's U statistic based on a pre-computed confusion matrix. Args: confmat: Confusion matrix for observed data Returns: Theil's U statistic rdevice)rrr rrZtensorr!)r Zs_xyrZp_xZs_xrrr_theils_u_computeQs r")rrrrr cCs.tt||g}t|||||}t|S)aSCompute `Theils Uncertainty coefficient`_ statistic measuring the association between two nominal data series. .. math:: U(X|Y) = \frac{H(X) - H(X|Y)}{H(X)} where :math:`H(X)` is entropy of variable :math:`X` while :math:`H(X|Y)` is the conditional entropy of :math:`X` given :math:`Y`. Theils's U is an asymmetric coefficient, i.e. :math:`TheilsU(preds, target) \neq TheilsU(target, preds)`. The output values lies in [0, 1]. 0 means y has no information about x while value 1 means y has complete information about x. Args: preds: 1D or 2D tensor of categorical (nominal) data - 1D shape: (batch_size,) - 2D shape: (batch_size, num_classes) target: 1D or 2D tensor of categorical (nominal) data - 1D shape: (batch_size,) - 2D shape: (batch_size, num_classes) nan_strategy: Indication of whether to replace or drop ``NaN`` values nan_replace_value: Value to replace ``NaN``s when ``nan_strategy = 'replace'`` Returns: Tensor containing Theil's U statistic Example: >>> from torchmetrics.functional.nominal import theils_u >>> _ = torch.manual_seed(42) >>> preds = torch.randint(10, (10,)) >>> target = torch.randint(10, (10,)) >>> theils_u(preds, target) tensor(0.8530) )lenrcatuniquerr")rrrrrr rrrtheils_uls)r&)matrixrrr c Cst|||jd}tj|||jd}tt|dD]r\}}|dd|f|dd|f}}tt ||g } t ||| ||} t | |||f<t | j |||f<q6|S)aCompute `Theil's U`_ statistic between a set of multiple variables. This can serve as a convenient tool to compute Theil's U statistic for analyses of correlation between categorical variables in your dataset. Args: matrix: A tensor of categorical (nominal) data, where: - rows represent a number of data points - columns represent a number of categorical (nominal) features nan_strategy: Indication of whether to replace or drop ``NaN`` values nan_replace_value: Value to replace ``NaN``s when ``nan_strategy = 'replace'`` Returns: Theil's U statistic for a dataset of categorical variables Example: >>> from torchmetrics.functional.nominal import theils_u_matrix >>> _ = torch.manual_seed(42) >>> matrix = torch.randint(0, 4, (200, 5)) >>> theils_u_matrix(matrix) tensor([[1.0000, 0.0202, 0.0142, 0.0196, 0.0353], [0.0202, 1.0000, 0.0070, 0.0136, 0.0065], [0.0143, 0.0070, 1.0000, 0.0125, 0.0206], [0.0198, 0.0137, 0.0125, 1.0000, 0.0312], [0.0352, 0.0065, 0.0204, 0.0308, 1.0000]]) r r rN)r rrZonesr! itertools combinationsranger#r$r%rr"T) r'rrZ num_variablesZtheils_u_matrix_valueijxyrr rrrtheils_u_matrixs  "r0)rr)rr)rr)r(typingrrrrZtyping_extensionsrZ7torchmetrics.functional.classification.confusion_matrixrZ%torchmetrics.functional.nominal.utilsrrr rintfloatrr"r&r0rrrrsF     0