diff options
Diffstat (limited to 'report/validation_data.rb')
-rw-r--r-- | report/validation_data.rb | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/report/validation_data.rb b/report/validation_data.rb index 3f96cc8..d87e491 100644 --- a/report/validation_data.rb +++ b/report/validation_data.rb @@ -3,6 +3,17 @@ VAL_ATTR_VARIANCE = [ :area_under_roc, :percent_correct, :root_mean_squared_error, :mean_absolute_error, :r_square, :accuracy ] VAL_ATTR_RANKING = [ :area_under_roc, :percent_correct, :true_positive_rate, :true_negative_rate, :accuracy ] +ATTR_NICE_NAME = {} + +class String + def nice_attr() + if ATTR_NICE_NAME.has_key?(self) + return ATTR_NICE_NAME[self] + else + return self.to_s.gsub(/_id$/, "").gsub(/_/, " ").capitalize + end + end +end class Object @@ -238,7 +249,7 @@ module Reports # def to_array(attributes, remove_nil_attributes=true, true_class_value=nil) array = Array.new - array.push(attributes) + array.push(attributes.collect{|a| a.to_s.nice_attr}) attribute_not_nil = Array.new(attributes.size) @validations.each do |v| index = 0 |