Skip to content

Commit 1f0b708

Browse files
committed
Use training set for training score.
1 parent 0b10a2e commit 1f0b708

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/test/groovy/net/zomis/machlearn/text/duga/ProgrammersCommentTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ public void commentLearning() {
9393
ClassifierFunction function = (theta, x) ->
9494
LogisticRegression.hypothesis(theta, x) >= 0.3;
9595
System.out.println("ALL Score: " + data.precisionRecallF1(learnedTheta, function));
96-
System.out.println("Training Score: " + data.precisionRecallF1(learnedTheta, function));
96+
System.out.println("Training Score: " + trainingSet.precisionRecallF1(learnedTheta, function));
9797
System.out.println("CrossVal Score: " + crossValidSet.precisionRecallF1(learnedTheta, function));
9898
System.out.println("TestSet Score: " + testSet.precisionRecallF1(learnedTheta, function));
9999

0 commit comments

Comments
 (0)