diff options
Diffstat (limited to 'lib/lazar.rb')
-rw-r--r-- | lib/lazar.rb | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/lib/lazar.rb b/lib/lazar.rb index 1e123d7..2bb89cd 100644 --- a/lib/lazar.rb +++ b/lib/lazar.rb @@ -170,7 +170,9 @@ module OpenTox # prepare prediction dataset prediction_dataset = LazarPrediction.new( :title => "Lazar prediction for #{prediction_feature.title}", - :creator => __FILE__ + :creator => __FILE__, + :prediction_feature_id => prediction_feature.id + ) confidence_feature = OpenTox::NumericFeature.find_or_create_by( "title" => "Prediction confidence" ) warning_feature = OpenTox::NominalFeature.find_or_create_by("title" => "Warnings") @@ -182,6 +184,14 @@ module OpenTox end end + + def training_dataset + Dataset.find training_dataset_id + end + + def prediction_feature + Feature.find prediction_feature_id + end def training_activities i = @training_dataset.feature_ids.index prediction_feature_id |