Add log message with best param

skylander86 · Sep 27, 2017 · 3619a4f · 3619a4f
1 parent 421b454
commit 3619a4f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ycml/classifiers/gridsearch.py b/ycml/classifiers/gridsearch.py
@@ -45,6 +45,7 @@ def fit_binarized(self, X_featurized, Y_binarized, validation_data=None, **kwarg
         param_scores = Parallel(n_jobs=self.n_jobs)(delayed(_fit_classifier)(klass, self.classifier_args, param, self.metric, X_train, Y_train, X_validation, Y_validation) for param in ParameterGrid(self.param_grid))
 
         best_param, best_score = max(param_scores, key=lambda x: x[1])
+        logger.info('Best scoring param is {} with score {}.'.format(best_param, best_score))
 
         classifier_args = {}
         classifier_args.update(self.classifier_args)