diff options
author | mguetlein <martin.guetlein@gmail.com> | 2011-08-18 11:25:39 +0200 |
---|---|---|
committer | mguetlein <martin.guetlein@gmail.com> | 2011-08-18 11:25:39 +0200 |
commit | 32f9e3f97c1a8278cf5022f619d9e969a37fed38 (patch) | |
tree | 236c84463266050703e25c716e799225eed74faa /report/report_factory.rb | |
parent | d27d53d98238ede80fc3b1a0c277ca890a84c736 (diff) |
fix ROC stuff [2]
Diffstat (limited to 'report/report_factory.rb')
-rwxr-xr-x | report/report_factory.rb | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/report/report_factory.rb b/report/report_factory.rb index 1cf7b94..e3e0f3c 100755 --- a/report/report_factory.rb +++ b/report/report_factory.rb @@ -78,10 +78,11 @@ module Reports::ReportFactory report.add_section("Plots") report.add_confidence_plot(validation_set) if (validation_set.get_accept_values.size == 2) - report.add_roc_plot(validation_set, validation_set.get_accept_values[0]) - else - validation_set.get_accept_values.each do |accept_value| - report.add_roc_plot(validation_set, accept_value) + 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]) end end report.end_section @@ -132,10 +133,11 @@ module Reports::ReportFactory report.add_confidence_plot(validation_set,nil,split_attribute) if (validation_set.get_accept_values.size == 2) - report.add_roc_plot(validation_set, validation_set.get_accept_values[0], split_attribute) - else - validation_set.get_accept_values.each do |accept_value| - report.add_roc_plot(validation_set, accept_value, split_attribute) + 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) end end end |