{"repo":"enochkan/torch-metrics","free":true,"listed":false,"github":"https://github.com/enochkan/torch-metrics","clone":"git clone https://github.com/enochkan/torch-metrics.git","description":"Metrics for model evaluation in pytorch","language":"Python","stars":110,"topics":["pytorch","machine-learning","deep-learning","metrics","pytorch-implementation","pytorch-cnn","computer-vision","torch-metrics"],"license":"MIT","category":"machine-learning","readme_excerpt":"Torch-metrics Model evaluation metrics for PyTorch Torch-metrics serves as a custom library to provide common ML evaluation metrics in Pytorch, similar to tf.keras.metrics . As summarized in this issue, Pytorch does not have a built-in libary torch.metrics for model evaluation metrics. This is similar to the metrics library in PyTorch Lightning. Usage - pip install --upgrade torch-metrics Implementation Metrics from tf.keras.metrics and other metrics that are already implemented vs to-do - [x] MeanSquaredError class - [x] RootMeanSquaredError class - [x] MeanAbsoluteError class - [x] Precision class - [x] Recall class - [x] MeanIoU class - [x] DSC class (Dice Similarity Coefficient) - [x] F1Score class - [x] RSquared class - [x] Hinge class - [x] SquaredHinge class - [x] LogCoshError class - [x] Accuracy class - [x] KLDivergence class - [ ] CosineSimilarity class - [ ] AUC class - [ ] BinaryCrossEntropy class - [ ] CategoricalCrossEntropy class - [ ] SparseCategoricalCrossentropy class Local Development and Testing To quickly get started with local development, run: To test, run: Pre-commit hooks To run pre-commit against all files: Contributing Please raise issues or feature requests here. It will be extremely helpful if you comment on a specific issue before working on it. This provides visibility for others who also intend to work on the same issue. Reference any pull requests to their original issues.","default_branch":null,"files":null,"tree":[],"storefront":"/r/enochkan","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/enochkan/torch-metrics/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}