From 85553b339acf3f9285a1c03b2fff342d9ddb9b6b Mon Sep 17 00:00:00 2001
From: Christoph Helma <helma@in-silico.ch>
Date: Wed, 11 Jan 2017 16:00:07 +0100
Subject: documentation for all classes

---
 lib/feature_selection.rb | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'lib/feature_selection.rb')

diff --git a/lib/feature_selection.rb b/lib/feature_selection.rb
index 65f9752..c596b1f 100644
--- a/lib/feature_selection.rb
+++ b/lib/feature_selection.rb
@@ -1,13 +1,16 @@
 module OpenTox
   module Algorithm
     
+    # Feature selection algorithms
     class FeatureSelection
 
+      # Select features correlated to the models prediction feature
+      # @param [OpenTox::Model::Lazar]
       def self.correlation_filter model
         relevant_features = {}
         R.assign "dependent", model.dependent_variables.collect{|v| to_r(v)}
         model.descriptor_weights = []
-        selected_variables = []
+        selected_variables = [] 
         selected_descriptor_ids = []
         model.independent_variables.each_with_index do |v,i|
           v.collect!{|n| to_r(n)}
-- 
cgit v1.2.3