摘要:
背景: sklearn是非常成熟、使用方便的机器学习算法工具包,但是其运行速度慢,只能支持CPU运行, 一直是一个诟病。 现在也出现了对sklearn中算法加速的工具包,比如thundersvm、sklearnex, 但是测试之后,加速效果十分有限。 下面提供NVIDIA开发基于GPU的cuML机器 阅读全文
摘要:
最近使用sklearn跑一些机器学习的实验对比,发现许多算法随着数据集增大,训练时间呈几何增加,加之交叉验证、参数选择等,非常耗时。 对此,已经有许多优化方案被提出。这里给出一个关于K-NN分类算法的快速实现工具推荐:基于faiss实现版本, 亲测速度提升明显 (1) 安装GPU版本faiss (h 阅读全文