diff options
| author | Martin Gütlein <martin.guetlein@gmail.com> | 2010-01-11 14:33:46 +0100 |
|---|---|---|
| committer | Martin Gütlein <martin.guetlein@gmail.com> | 2010-01-11 14:33:46 +0100 |
| commit | 86f056bbb8f8b083313a8dd55b1b8aa47574afe1 (patch) | |
| tree | f96a2d882da5a8b586a15468d9bf07ca7e079007 /report/xml_report_util.rb | |
| parent | bc6a5145c842d9e47a0f36955d87484ac2f19061 (diff) | |
adjusted reporting to recent api changes
Diffstat (limited to 'report/xml_report_util.rb')
| -rw-r--r-- | report/xml_report_util.rb | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/report/xml_report_util.rb b/report/xml_report_util.rb index 06e37dd..ce17fde 100644 --- a/report/xml_report_util.rb +++ b/report/xml_report_util.rb @@ -15,7 +15,7 @@ module Reports::XMLReportUtil num_classes = Math.sqrt(confusion_matrix.size) class_values = [] - confusion_matrix.each{ |key_map,value| class_values.push(key_map[:actual]) if class_values.index(key_map[:actual])==nil } + confusion_matrix.each{ |key_map,value| class_values.push(key_map[:confusion_matrix_actual]) if class_values.index(key_map[:confusion_matrix_actual])==nil } raise "confusion matrix invalid "+confusion_matrix.inspect unless num_classes.to_i == num_classes and class_values.size == num_classes sum_predicted = {} @@ -24,8 +24,8 @@ module Reports::XMLReportUtil sum_pred = 0 sum_act = 0 confusion_matrix.each do |key_map,value| - sum_pred += value if key_map[:predicted]==class_value - sum_act += value if key_map[:actual]==class_value + sum_pred += value if key_map[:confusion_matrix_predicted]==class_value + sum_act += value if key_map[:confusion_matrix_actual]==class_value end sum_predicted[class_value] = sum_pred sum_actual[class_value] = sum_act @@ -38,7 +38,7 @@ module Reports::XMLReportUtil class_values.each do |predicted| row = [ (confusion.size==2 ? "predicted" : ""), predicted ] class_values.each do |actual| - row.push( confusion_matrix[{:actual => actual, :predicted => predicted}].to_nice_s ) + row.push( confusion_matrix[{:confusion_matrix_actual => actual, :confusion_matrix_predicted => predicted}].to_nice_s ) end row.push( sum_predicted[predicted].to_nice_s ) confusion.push( row ) |