diff options
author | mguetlein <martin.guetlein@gmail.com> | 2011-08-18 12:57:33 +0200 |
---|---|---|
committer | mguetlein <martin.guetlein@gmail.com> | 2011-08-18 12:57:33 +0200 |
commit | bad2d7444ab40a59770678c0b0e4057d5edeceef (patch) | |
tree | 1de89743e3d1a92b220295a86f951969afdea661 /report/report_factory.rb | |
parent | 32f9e3f97c1a8278cf5022f619d9e969a37fed38 (diff) |
add confidence plots for various classification stats
Diffstat (limited to 'report/report_factory.rb')
-rwxr-xr-x | report/report_factory.rb | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/report/report_factory.rb b/report/report_factory.rb index e3e0f3c..2a50869 100755 --- a/report/report_factory.rb +++ b/report/report_factory.rb @@ -76,15 +76,21 @@ module Reports::ReportFactory report.add_result(validation_set, [:validation_uri] + VAL_ATTR_TRAIN_TEST + VAL_ATTR_CLASS, "Results", "Results") report.add_confusion_matrix(val) report.add_section("Plots") - report.add_confidence_plot(validation_set) if (validation_set.get_accept_values.size == 2) if validation_set.get_true_accept_value!=nil report.add_roc_plot(validation_set, validation_set.get_true_accept_value) else report.add_roc_plot(validation_set, validation_set.get_accept_values[0]) report.add_roc_plot(validation_set, validation_set.get_accept_values[1]) + report.align_last_two_images "ROC Plots" end end + report.add_confidence_plot(validation_set) + validation_set.get_accept_values.each do |accept_value| + report.add_confidence_plot(validation_set, accept_value, nil) + report.add_confidence_plot(validation_set, nil, accept_value) + report.align_last_two_images "Confidence Plots" + end report.end_section when "regression" report.add_result(validation_set, [:validation_uri] + VAL_ATTR_TRAIN_TEST + VAL_ATTR_REGR, "Results", "Results") @@ -131,15 +137,21 @@ module Reports::ReportFactory report.add_section("Plots") [nil, :crossvalidation_fold].each do |split_attribute| - report.add_confidence_plot(validation_set,nil,split_attribute) if (validation_set.get_accept_values.size == 2) if validation_set.get_true_accept_value!=nil report.add_roc_plot(validation_set, validation_set.get_true_accept_value,split_attribute) else report.add_roc_plot(validation_set, validation_set.get_accept_values[0],split_attribute) report.add_roc_plot(validation_set, validation_set.get_accept_values[1],split_attribute) + report.align_last_two_images "ROC Plots" end end + report.add_confidence_plot(validation_set,nil,nil,split_attribute) + validation_set.get_accept_values.each do |accept_value| + report.add_confidence_plot(validation_set, accept_value, nil,split_attribute) + report.add_confidence_plot(validation_set, nil, accept_value,split_attribute) + report.align_last_two_images "Confidence Plots" + end end report.end_section report.add_result(validation_set, [:validation_uri, :validation_report_uri]+VAL_ATTR_CV+VAL_ATTR_CLASS-[:num_folds, :dataset_uri, :algorithm_uri], |