Apache
/
lucene
mirror of https://github.com/apache/lucene.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

LUCENE-6854 - adjusted precision calculation, minor fix in SNBC test 

git-svn-id: https://svn.apache.org/repos/asf/lucene/dev/trunk@1723759 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Tommaso Teofili 2016-01-08 16:49:48 +00:00
parent 2580956b8b
commit 056940632d
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lucene/classification/src/java/org/apache/lucene/classification/utils/ConfusionMatrixGenerator.java
View File

@ -167,7 +167,7 @@ public class ConfusionMatrixGenerator {
public double getPrecision(String klass) { public double getPrecision(String klass) {
Map<String, Long> classifications = linearizedMatrix.get(klass); Map<String, Long> classifications = linearizedMatrix.get(klass);
double tp = 0; double tp = 0;
double fp = 0; double fp = -1;
if (classifications != null) { if (classifications != null) {
for (Map.Entry<String, Long> entry : classifications.entrySet()) { for (Map.Entry<String, Long> entry : classifications.entrySet()) {
if (klass.equals(entry.getKey())) { if (klass.equals(entry.getKey())) {
@ -253,7 +253,7 @@ public class ConfusionMatrixGenerator {
*/ */
public double getPrecision() { public double getPrecision() {
double tp = 0; double tp = 0;
double fp = 0; double fp = -linearizedMatrix.size();
for (Map.Entry<String, Map<String, Long>> classification : linearizedMatrix.entrySet()) { for (Map.Entry<String, Map<String, Long>> classification : linearizedMatrix.entrySet()) {
String klass = classification.getKey(); String klass = classification.getKey();
for (Map.Entry<String, Long> entry : classification.getValue().entrySet()) { for (Map.Entry<String, Long> entry : classification.getValue().entrySet()) {

3
lucene/classification/src/test/org/apache/lucene/classification/SimpleNaiveBayesClassifierTest.java
View File

@ -75,7 +75,8 @@ public class SimpleNaiveBayesClassifierTest extends ClassificationTestBase<Bytes
try { try {
Analyzer analyzer = new NGramAnalyzer(); Analyzer analyzer = new NGramAnalyzer();
leafReader = getSampleIndex(analyzer); leafReader = getSampleIndex(analyzer);
checkCorrectClassification(new CachingNaiveBayesClassifier(leafReader, analyzer, null, categoryFieldName, textFieldName), TECHNOLOGY_INPUT, TECHNOLOGY_RESULT); SimpleNaiveBayesClassifier classifier = new SimpleNaiveBayesClassifier(leafReader, analyzer, null, categoryFieldName, textFieldName);
checkCorrectClassification(classifier, TECHNOLOGY_INPUT, TECHNOLOGY_RESULT);
} finally { } finally {
if (leafReader != null) { if (leafReader != null) {
leafReader.close(); leafReader.close();