From 7b2d98ed2c33aad819d8226e04cc59bf2bf86eaa Mon Sep 17 00:00:00 2001
From: Fabian Pedregosa <fabian.pedregosa@inria.fr>
Date: Fri, 26 Nov 2010 17:36:48 +0100
Subject: [PATCH] Fix precompute keyword in LARS.
---
scikits/learn/linear_model/lars.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/scikits/learn/linear_model/lars.py b/scikits/learn/linear_model/lars.py
index bc9ed86653..fc8cd1c034 100644
--- a/scikits/learn/linear_model/lars.py
+++ b/scikits/learn/linear_model/lars.py
@@ -358,12 +358,12 @@ class LARS(LinearModel):
X[:, nonzeros] /= norms[nonzeros]
# precompute if n_samples > n_features
- if precompute == True or \
- (precompute == 'auto' and X.shape[0] > X.shape[1]):
- Gram = np.dot(X.T, X)
- elif hasattr(precompute, '__array__'):
+ if hasattr(precompute, '__array__'):
# copy as it's going to be modified
Gram = precompute.copy()
+ elif precompute == True or \
+ (precompute == 'auto' and X.shape[0] > X.shape[1]):
+ Gram = np.dot(X.T, X)
else:
Gram = None
--
GitLab