[OPENNLP-1187] Issue in finding accuracy of model

• Type: Bug
• Status: Closed
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 1.8.4
• Fix Version/s: 2.1.1
• Component/s: Doccat, Machine Learning
• Flags: Important
• Assignee: Martin Wiesner
• Reporter: Aman Garg
• Created: 26/Feb/18 18:44
• Updated: 28/Jan/23 15:00
• Resolved: 15/Dec/22 07:10

Description

the trainingStats function in NaiveBayesTrainer class is not working properly and display wrong result.

In findParameters(), at line 154 i.e. 

EvalParameters evalParams = new EvalParameters(params, numOutcomes);

should be replaced by following block:

 

double[] outcomeTotals = new double[outcomeLabels.length];

    for (int i = 0; i < params.length; ++i) {

      Context context = params[i];

      for (int j = 0; j < context.getOutcomes().length; ++j)

{         int outcome = context.getOutcomes()[j];         double count = context.getParameters()[j];         outcomeTotals[outcome] += count;       }

    }

evalParams = new NaiveBayesEvalParameters(params,

outcomeLabels.length, outcomeTotals, predLabels.length);