Author: tdunning
Date: Wed Sep 8 18:02:25 2010
New Revision: 995174
URL: http://svn.apache.org/viewvc?rev=995174&view=rev
Log:
Make log-likelihood behavior better in the corners.
Modified:
mahout/trunk/core/src/main/java/org/apache/mahout/classifier/sgd/CrossFoldLearner.java
Modified:
mahout/trunk/core/src/main/java/org/apache/mahout/classifier/sgd/CrossFoldLearner.java
URL:
http://svn.apache.org/viewvc/mahout/trunk/core/src/main/java/org/apache/mahout/classifier/sgd/CrossFoldLearner.java?rev=995174&r1=995173&r2=995174&view=diff
==============================================================================
---
mahout/trunk/core/src/main/java/org/apache/mahout/classifier/sgd/CrossFoldLearner.java
(original)
+++
mahout/trunk/core/src/main/java/org/apache/mahout/classifier/sgd/CrossFoldLearner.java
Wed Sep 8 18:02:25 2010
@@ -19,7 +19,10 @@ import java.util.List;
* record should be passed with each training example.
*/
public class CrossFoldLearner extends AbstractVectorClassifier implements
OnlineLearner {
- private int record;
+ // minimum score to be used for computing log likelihood
+ private static final double MIN_SCORE = 1e-50;
+
+ private int record = 0;
private OnlineAuc auc = new OnlineAuc();
private double logLikelihood;
private final List<OnlineLogisticRegression> models = Lists.newArrayList();