Machine learning metrics for distributed, scalable PyTorch applications. - torchmetrics/src/torchmetrics/clustering/mutual_info_score.py at v1.3.2 · Lightning-AI/torchmetrics
Sometimes docs build are failing due to a bad sampling in clustering metrics: https://github.com/Lightning-AI/torchmetrics/actions/runs/11342203643/job/31542759627 This should make it more robust by increasing the number of samples. Before submitting PR review Anyone in the community is free to ...