scikit-learn-compatible

标签

Cards List
#scikit-learn-compatible

TorchKM:面向GPU的核学习与模型选择库

arXiv cs.LG · 2026-06-08 缓存

TorchKM是一个开源的GPU加速核机器库(支持向量机、核逻辑回归等),采用scikit-learn风格的API。通过重用矩阵运算加速训练和模型选择,相比标准基线实现了显著的加速比。

0 人收藏 0 人点赞
← 返回首页

提交意见反馈