torch-metrics

Metrics library

A collection of common machine learning evaluation metrics implemented in PyTorch

Metrics for model evaluation in pytorch

GitHub

110 stars
3 watching
14 forks
Language: Python
last commit: over 3 years ago
computer-visiondeep-learningmachine-learningmetricspytorchpytorch-cnnpytorch-implementationtorch-metrics

Related projects:

Repository Description Stars
oml-team/open-metric-learning A PyTorch-based framework for training and validating models that produce high-quality embeddings for computer vision and other tasks. 882
benhamner/metrics Provides implementations of various supervised machine learning evaluation metrics in multiple programming languages. 1,627
ahmedfgad/torchga Trains PyTorch models using the Genetic Algorithm 95
lartpang/pysodmetrics A library providing an implementation of various metrics for object segmentation and saliency detection in computer vision. 144
pytorchbearer/torchbearer A PyTorch model fitting library designed to simplify the process of training deep learning models. 636
torchdr/torchdr A PyTorch-based library for dimensionality reduction and manifold learning using affinity matrices. 68
yu-changqian/torchseg A toolkit for building and training semantic segmentation models using PyTorch. 1,408
torch/cutorch Provides a CUDA backend for the PyTorch deep learning framework 336
nearai/torchfold A PyTorch module for dynamic batching and optimized computation on deep neural networks 221
vermeille/torchelie A collection of utility functions and tools for building deep learning models with PyTorch 111
kakaobrain/torchgpipe A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients. 817
fanq15/caffe_to_torch_to_pytorch A tool for converting models between Caffe and PyTorch 149
inisis/brocolli Converts PyTorch models to various formats for deployment and testing in deep learning frameworks. 341
pascaldekloe/metrics Provides a simple and efficient way to track and expose performance metrics in Go applications. 28
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432