From 05ef0c4abed2e43f816ddc2b5524b04fcfb2df17 Mon Sep 17 00:00:00 2001 From: Rishabh Gupta Date: Sat, 17 Jun 2017 14:45:58 -0400 Subject: [PATCH] further changes for stall count --- few/tests/test_few.py | 9 --------- few/variation.py | 2 +- 2 files changed, 1 insertion(+), 10 deletions(-) diff --git a/few/tests/test_few.py b/few/tests/test_few.py index cca752e..68aded4 100644 --- a/few/tests/test_few.py +++ b/few/tests/test_few.py @@ -67,15 +67,6 @@ def test_few_with_parents_weight(): few_score = learner.score(features[:300], target[:300]) test_score = learner.score(features[300:],target[300:]) - lasso = LassoLarsCV() - lasso.fit(learner._training_features,learner._training_labels) - lasso_score = lasso.score(features[:300], target[:300]) - print("few score:",few_score,"lasso score:",lasso_score) - print("few test score:",test_score,"lasso test score:",lasso.score(features[300:],target[300:])) - assert few_score >= lasso_score - - print("lasso coefficients:",lasso.coef_) - def test_few_at_least_as_good_as_default(): """test_few.py: few performs at least as well as the default ML """ diff --git a/few/variation.py b/few/variation.py index adac873..0d9f437 100644 --- a/few/variation.py +++ b/few/variation.py @@ -31,7 +31,7 @@ def variation(self,parents): elif hasattr(self.ml,'feature_importances_'): # for tree methods, filter our individuals with 0 feature importance if self.weight_parents: - weights = abs(self.ml.feature_importances_) + weights = self.ml.feature_importances_ weights = weights/sum(weights) offspring = copy.deepcopy(list(np.random.choice(self.valid(parents), self.population_size, p=weights))) else: