From c87b0d6866acb246d35fc457805340aef28147b8 Mon Sep 17 00:00:00 2001 From: mguetlein Date: Fri, 17 Jun 2011 16:19:11 +0200 Subject: fix chart keys (removing 'default', adding crossvalidation fold) --- report/plot_factory.rb | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/report/plot_factory.rb b/report/plot_factory.rb index 78d2e05..d6c7d35 100644 --- a/report/plot_factory.rb +++ b/report/plot_factory.rb @@ -106,6 +106,7 @@ module Reports attribute_values.each do |value| begin data << transform_roc_predictions(validation_set.filter({split_set_attribute => value}), class_value, false ) + data[-1].name = split_set_attribute.to_s.nice_attr+" "+value.to_s rescue LOGGER.warn "could not create ROC plot for "+value.to_s end @@ -133,7 +134,7 @@ module Reports attribute_values.each do |value| begin data = transform_confidence_predictions(validation_set.filter({split_set_attribute => value}), class_value, false) - names << value.to_s + names << split_set_attribute.to_s.nice_attr+" "+value.to_s confidence << data[:confidence][0] performance << data[:performance][0] rescue @@ -298,7 +299,7 @@ module Reports tp_fp_rates[:youden].each do |point,confidence| labels << ["confidence: "+confidence.to_nice_s, point[0], point[1]] end if add_label - RubyPlot::LinePlotData.new(:name => "default", :x_values => tp_fp_rates[:fp_rate], :y_values => tp_fp_rates[:tp_rate], :labels => labels) + RubyPlot::LinePlotData.new(:name => "", :x_values => tp_fp_rates[:fp_rate], :y_values => tp_fp_rates[:tp_rate], :labels => labels) end @@ -337,7 +338,7 @@ module Reports else confidence_values = validation_set.validations[0].get_predictions.get_prediction_values(class_value) pref_conf_rates = get_performance_confidence_rates(confidence_values, validation_set.unique_feature_type) - return { :names => ["default"], :performance => [pref_conf_rates[:performance]], :confidence => [pref_conf_rates[:confidence]] } + return { :names => [""], :performance => [pref_conf_rates[:performance]], :confidence => [pref_conf_rates[:confidence]] } end end -- cgit v1.2.3