From 1d1348eb2126a0a8ee1a18b37a5254b59c3a4c76 Mon Sep 17 00:00:00 2001 From: Olivier Grisel Date: Fri, 30 Dec 2011 10:00:39 +0100 Subject: [PATCH] Update random forest face example to use several cores --- examples/ensemble/plot_forest_importances_faces.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/examples/ensemble/plot_forest_importances_faces.py b/examples/ensemble/plot_forest_importances_faces.py index 9645a62a1848d..29f89a5455931 100644 --- a/examples/ensemble/plot_forest_importances_faces.py +++ b/examples/ensemble/plot_forest_importances_faces.py @@ -9,11 +9,15 @@ """ print __doc__ +from time import time import pylab as pl from sklearn.datasets import fetch_olivetti_faces from sklearn.ensemble import ExtraTreesClassifier +# Number of cores to use to perform parallel fitting of the forest model +n_jobs=2 + # Loading the digits dataset data = fetch_olivetti_faces() X = data.images.reshape((len(data.images), -1)) @@ -24,12 +28,15 @@ y = y[mask] # Build a forest and compute the pixel importances +print "Fitting ExtraTreesClassifier on faces data with %d cores..." % n_jobs +t0 = time() forest = ExtraTreesClassifier(n_estimators=1000, max_features=128, compute_importances=True, - random_state=0) + random_state=0, n_jobs=n_jobs) forest.fit(X, y) +print "done in %0.3fs" % (time() - t0) importances = forest.feature_importances_ importances = importances.reshape(data.images[0].shape)