面向深度序分类的 Python 包 dlordinal
机器学习
2025-01-13 v3
摘要
dlordinal 是一个新的 Python 库，统一了文献中可用于深度序分类的各种最新方法。该库使用 PyTorch 作为底层框架，实现了针对序分类问题的最佳性能的 SOTA 深度学习技术。序方法旨在利用目标变量中存在的排序信息。 Specifically, 它包括损失函数、各种输出层、dropout 技术、软标签方法以及其他适当设计以纳入序信息的分类策略。此外，由于序分类中新方案的性能指标依赖于目标与预测类别在序尺度上的距离，因此也包括适合的序评估指标。dlordinal 在 BSD-3-Clause 许可证下分发，可在 https://github.com/ayrna/dlordinal 获取。
引用
@article{arxiv.2407.17163,
title = {dlordinal: a Python package for deep ordinal classification},
author = {Francisco Bérchez-Moreno and Víctor M. Vargas and Rafael Ayllón-Gavilán and David Guijo-Rubio and César Hervás-Martínez and Juan C. Fernández and Pedro A. Gutiérrez},
journal= {arXiv preprint arXiv:2407.17163},
year = {2025}
}