张量流和sklearn的速度差异

时间:2019-02-26 03:52:19

标签: python tensorflow scikit-learn

我正在从事一个需要实时反馈的项目。该项目是使用tensorflow构建的,它需要使用tensorflow语言编写的所有算法。

现在的问题是,我需要使用k-means,在sklearn中,它使用起来很简洁,并且需要33 milliseconds来响应。但是,当移至tensorflow时,我尝试了contrib.kmeansClustering,它相对较复杂,我需要手动设置num_iters,并选择了13s,这对项目是不容许的。 / p>

我想知道,尽管使用neural net中的tensorflow很方便,但在实现通用tensorflow算法中,sklearnmachine learning慢得多吗?如果我放弃contrib并从头开始在tensorflow中实施,它会和在sklearn中一样快吗?

0 个答案:

没有答案