Random Forests for the GPU using PyCUDA
CudaTree is an implementation of Leo Breiman’s Random Forests adapted to run on the GPU. A random forest is an ensemble of randomized decision trees which vote together to predict new labels. CudaTree parallelizes the construction of each individual tree in the ensemble and thus is able to train faster than the latest version of 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 is writen for Python 2.7 and depends on:
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size||File type||Python version||Upload date||Hashes|
|Filename, size cudatree-0.6.tar.gz (22.8 kB)||File type Source||Python version None||Upload date||Hashes View hashes|