diff options
Diffstat (limited to 'lib/model.rb')
-rw-r--r-- | lib/model.rb | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/lib/model.rb b/lib/model.rb index 36011a0..547144f 100644 --- a/lib/model.rb +++ b/lib/model.rb @@ -138,16 +138,19 @@ module OpenTox end class LazarFminerClassification < LazarClassification - def self.create training_dataset + field :feature_calculation_parameters, type: Hash + + def self.create training_dataset, fminer_params={} model = super(training_dataset) model.update "_type" => self.to_s # adjust class model = self.find model.id # adjust class model.neighbor_algorithm = "OpenTox::Algorithm::Neighbor.fminer_similarity" model.neighbor_algorithm_parameters = { :feature_calculation_algorithm => "OpenTox::Algorithm::Descriptor.smarts_match", - :feature_dataset_id => Algorithm::Fminer.bbrc(training_dataset).id, + :feature_dataset_id => Algorithm::Fminer.bbrc(training_dataset,fminer_params).id, :min_sim => 0.3 } + model.feature_calculation_parameters = fminer_params model.save model end |