torchcriterion 0.0.1

A modular PyTorch loss function library with popular criteria for classification, regression, segmentation, and metric learning.

torchcriterion

torchcriterion is a modular, extensible library of PyTorch-compatible loss functions ("criteria") for classification, regression, segmentation, and metric learning tasks. It offers a curated collection of both standard and custom loss functions, built with flexibility and composability in mind.

---

🚀 Features

• 🧱 Modular architecture for clean API and extension
• 🧪 Ready-to-use implementations of popular losses
• 🧩 Supports multi-loss composition and custom scheduling
• ⚡ Fully compatible with PyTorch’s autograd and GPU acceleration

---

📦 Installation

pip install torchcriterion  # Coming soon to PyPI

---

🧰 Supported Losses

Classification

• CrossEntropyLoss
• FocalLoss

Regression

• MSELoss
• HuberLoss

Segmentation

• DiceLoss
• TverskyLoss

Metric Learning

• TripletLoss
• ContrastiveLoss

---

🧪 Example Usage

from torchcriterion import FocalLoss

criterion = FocalLoss(gamma=2.0, alpha=0.25)
loss = criterion(predictions, targets)

---

📁 Project Structure

torchcriterion/
├── classification/
│   ├── cross_entropy.py
│   ├── focal.py
├── regression/
│   ├── mse.py
│   ├── huber.py
├── segmentation/
│   ├── dice.py
│   ├── tversky.py
├── metric_learning/
│   ├── triplet.py
│   ├── contrastive.py
├── base.py
├── __init__.py

---

📜 License

This project is licensed under the MIT License. See the LICENSE file for details.

---

🙌 Contributing

Pull requests, ideas, and issues are welcome! Feel free to open a PR or start a discussion.

---

👤 Author

Developed by TransformerTitan — @TransformerTitan

---

⭐️ Star the Repo

If you find this library useful, please consider starring it to show your support!

---

🔗 Related Projects

• torchmetrics — for evaluation metrics
• timm — for models with built-in loss support

---

Made with ❤️ and PyTorch