使用pycuda的gpu的随机林
cudatree的Python项目详细描述
CudaTree是Leo Breiman的随机林的一个实现,适合在GPU上运行随机森林是随机决策树的集合,它们一起投票来预测新的标签。cudatree并行化了集成中每个树的构造,因此能够比最新版本的scikits learn更快地进行训练。
用法
import numpy as np from cudatree import load_data, RandomForestClassifier x_train, y_train = load_data("digits") forest = RandomForestClassifier(n_estimators = 50, max_features = 6) forest.fit(x_train, y_train) forest.predict(x_train)
依赖关系
cudatree是为python 2.7编写的,取决于:
- SCIKIT学习
- 努比
- 皮库达
- 鹦鹉