gpt4 book ai didi

python - scikit-learn 适合大数据任务吗?

转载 作者:IT老高 更新时间:2023-10-28 21:07:50 26 4
gpt4 key购买 nike

我正在处理一项涉及使用机器学习技术的 TREC 任务,其中数据集包含超过 5 TB 的网络文档,计划从中提取词袋向量。 scikit-learn 有一组不错的功能,似乎符合我的需要,但我不知道它是否能很好地扩展以处理大数据。例如,HashingVectorizer 是否能够处理 5 TB 的文档,并行化是否可行?此外,对于大规模机器学习任务,还有哪些替代方案?

最佳答案

HashingVectorizer 如果您将数据迭代地分 block 成批处理,例如适合内存的 10k 或 100k 个文档,那么

HashingVectorizer 将起作用。

然后您可以将这批转换后的文档传递给支持 partial_fit 方法的线性分类器(例如 SGDClassifierPassiveAggressiveClassifier),然后迭代新批处理。

您可以开始在保留的验证集(例如 10k 个文档)上对模型进行评分,以监控部分训练模型的准确性,而无需等待查看所有样本。

您也可以在多台机器上对数据分区并行执行此操作,然后对生成的 coef_intercept_ 属性进行平均以获得所有数据的最终线性模型数据集。

我在 2013 年 3 月的 PyData 演讲中讨论了这个问题:http://vimeo.com/63269736

tutorial on paralyzing scikit-learn with IPython.parallel 中也有示例代码取自:https://github.com/ogrisel/parallel_ml_tutorial

关于python - scikit-learn 适合大数据任务吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/17017878/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com