Added feature title in result dataset (for ToxPredict)

1 files changed, 2 insertions, 0 deletions

diff --git a/lib/model.rb b/lib/model.rb
index 0f1cc22..1ff9df1 100644
--- a/lib/model.rb
+++ b/lib/model.rb
@@ -239,6 +239,8 @@ module OpenTox
             @prediction_dataset.add @compound.uri, value_feature_uri, prediction[:prediction]
           end
           @prediction_dataset.add @compound.uri, confidence_feature_uri, prediction[:confidence]
+          @prediction_dataset.features[value_feature_uri][DC.title] = @prediction_dataset.metadata[DC.title]
+          @prediction_dataset.features[confidence_feature_uri][DC.title] = "Confidence"
 
           if verbose
             if @feature_calculation_algorithm == "Substructure.match"