diff options
author | Martin Gütlein <martin.guetlein@gmail.com> | 2010-02-23 15:01:26 +0100 |
---|---|---|
committer | Martin Gütlein <martin.guetlein@gmail.com> | 2010-02-23 15:01:26 +0100 |
commit | 5c62857e2508efd5001ca0903f8afa49c85904a2 (patch) | |
tree | 84b94e31ec275b56e24a64c2f9638508dca9eb24 /report/report_factory.rb | |
parent | 98c03ab8e7631977fb15bb35a035a150e04f61ed (diff) |
disable validation logger init, add new roc plot functionality
Diffstat (limited to 'report/report_factory.rb')
-rw-r--r-- | report/report_factory.rb | 20 |
1 files changed, 11 insertions, 9 deletions
diff --git a/report/report_factory.rb b/report/report_factory.rb index b13624f..4331660 100644 --- a/report/report_factory.rb +++ b/report/report_factory.rb @@ -51,9 +51,10 @@ module Reports::ReportFactory if (val.classification?) report.add_section_result(validation_set, VAL_ATTR_TRAIN_TEST + VAL_ATTR_CLASS, "Results", "Results") - val.get_prediction_feature_values.each do |class_value| - report.add_section_roc_plot(validation_set, class_value, nil, "roc-plot-"+class_value+".svg") - end + report.add_section_roc_plot(validation_set, nil, nil, "roc-plot.svg") + #val.get_prediction_feature_values.each do |class_value| + #report.add_section_roc_plot(validation_set, class_value, nil, "roc-plot-"+class_value+".svg") + #end report.add_section_confusion_matrix(validation_set.first) else #regression report.add_section_result(validation_set, VAL_ATTR_TRAIN_TEST + VAL_ATTR_REGR, "Results", "Results") @@ -82,9 +83,11 @@ module Reports::ReportFactory if (validation_set.validations[0].percent_correct!=nil) #classification report.add_section_result(merged, VAL_ATTR_CV+VAL_ATTR_CLASS-[:crossvalidation_fold],"Mean Results","Mean Results") - validation_set.validations[0].get_prediction_feature_values.each do |class_value| - report.add_section_roc_plot(validation_set, class_value, nil, "roc-plot-"+class_value+".svg") - end + + report.add_section_roc_plot(validation_set, nil, nil, "roc-plot.svg") + #validation_set.validations[0].get_prediction_feature_values.each do |class_value| + #report.add_section_roc_plot(validation_set, class_value, nil, "roc-plot-"+class_value+".svg") + #end report.add_section_confusion_matrix(merged.first) report.add_section_result(validation_set, VAL_ATTR_CV+VAL_ATTR_CLASS-[:num_folds], "Results","Results") else #regression @@ -208,11 +211,10 @@ class Reports::ReportContent section_title="Confusion Matrix", section_text="This section contains the confusion matrix.", table_title="Confusion Matrix") - section_confusion = @xml_report.add_section(xml_report.get_root_element, section_title) @xml_report.add_paragraph(section_confusion, section_text) if section_text @xml_report.add_table(section_confusion, table_title, - Reports::XMLReportUtil::create_confusion_matrix( validation.confusion_matrix), false) + Reports::XMLReportUtil::create_confusion_matrix( validation.confusion_matrix ), false) end def add_section_roc_plot( validation_set, @@ -238,7 +240,7 @@ class Reports::ReportContent begin plot_file_path = add_tmp_file(plot_file_name) - Reports::RPlotFactory.create_roc_plot( plot_file_path, validation_set, class_value, split_set_attribute, validation_set.size>1 ) + Reports::PlotFactory.create_roc_plot( plot_file_path, validation_set, class_value, split_set_attribute, validation_set.size>1 ) @xml_report.add_imagefigure(section_roc, image_title, plot_file_name, "SVG", image_caption) rescue RuntimeError => ex LOGGER.error("could not create roc plot: "+ex.message) |