From a54db46684680d98311631804eca367cc949a715 Mon Sep 17 00:00:00 2001
From: Christoph Helma <helma@in-silico.ch>
Date: Tue, 26 Mar 2013 10:56:04 +0100
Subject: code cleanup and refactoring.

---
 lib/feature.rb | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 lib/feature.rb

(limited to 'lib/feature.rb')

diff --git a/lib/feature.rb b/lib/feature.rb
new file mode 100644
index 0000000..5d3d962
--- /dev/null
+++ b/lib/feature.rb
@@ -0,0 +1,37 @@
+module OpenTox
+
+  class Feature
+
+    # Find out feature type
+    # Classification takes precedence
+    # @return [String] Feature type
+    def feature_type
+      if self[RDF.type].include?(RDF::OT.NominalFeature)
+        "classification"
+      elsif self[RDF.type].include?(RDF::OT.NumericFeature)
+        "regression"
+      else
+        "unknown"
+      end
+    end
+
+    # Get accept values
+    # 
+    # @return[Array] Accept values
+    def accept_values
+      self[RDF::OT.acceptValue] ? self[RDF::OT.acceptValue].sort : nil
+    end
+
+    # Create value map
+    # @param [OpenTox::Feature] Feature
+    # @return [Hash] A hash with keys 1...feature.training_classes.size and values training classes
+    def value_map
+      unless defined? @value_map
+        accept_values ? @value_map = accept_values.each_index.inject({}) { |h,idx| h[idx+1]=accept_values[idx]; h } : @value_map = nil
+      end
+      @value_map
+    end
+
+  end
+
+end
-- 
cgit v1.2.3