相关项目
torchmetrics
TorchMetrics提供超过100种PyTorch指标的实现，拥有易用的API和支持自定义指标的功能，优化分布式训练并减少样板代码。它支持自动批次累积和设备同步，模块化设计使得指标可以自动放置在正确的设备上，并兼容PyTorch Lightning。覆盖音频、分类、检测、图像等多个领域，并提供可视化工具，便于理解机器学习算法。
stella-large-zh-v3-1792d
stella-large-zh-v3-1792d项目专注于文本相似性与分类问题，结合多种评估任务和数据集，如STS、分类、聚类、重排序、检索等，展现了其在中文自然语言处理中的强大能力。在MTEB评测集的不同任务下，该模型表现良好。在相似度评估中，Pearson和Spearman相关系数较高，而在分类任务中，模型的准确率和F1值均有提升。同时，在搜索和重排序任务中，其平均精度和召回率也表现不俗，使其成为适用于多种语言处理场景的工具。