lib/feature_selection.rb in lazar-1.0.0 vs lib/feature_selection.rb in lazar-1.0.1
- old
+ new
@@ -1,14 +1,17 @@
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)}
R.assign "independent", v
begin