diff options
-rw-r--r-- | lib/algorithm.rb | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/algorithm.rb b/lib/algorithm.rb index fb14482..f57954d 100644 --- a/lib/algorithm.rb +++ b/lib/algorithm.rb @@ -505,7 +505,7 @@ module OpenTox subsets = subsets[subsets>1] # Recursive feature elimination - rfProfile = rfe( x=features, y=y, rfeControl=rfeControl(functions=rfFuncs, number=50), sizes=subsets) + rfProfile = rfe( x=features, y=y, rfeControl=rfeControl(functions=rfFuncs, number=150), sizes=subsets) # read existing dataset and select most useful features csv=feats[,c("SMILES", rfProfile$optVariables)] |