Merge branch 'new_model_prediction_format' into development

1 files changed, 4 insertions, 5 deletions

diff --git a/lazar.rb b/lazar.rb
index ec055ee..0c91179 100644
--- a/lazar.rb
+++ b/lazar.rb
@@ -155,11 +155,10 @@ post '/lazar/?' do
     lazar.metadata[OT.dependentVariables] = prediction_feature.uri
     lazar.metadata[OT.trainingDataset] = dataset_uri
 		lazar.metadata[OT.featureDataset] = feature_dataset_uri
-
-    if prediction_feature.feature_type == "classification"
-      lazar.metadata[RDF.type] = [OTA.ClassificationLazySingleTarget]
-    elsif prediction_feature.feature_type == "regression"
-      lazar.metadata[RDF.type] = [OTA.RegressionLazySingleTarget]
+    if training_activities.feature_type.to_s == "classification"
+      lazar.metadata[RDF.type] = [OT.Model, OTA.ClassificationLazySingleTarget]
+    elsif training_activities.feature_type.to_s == "regression"
+      lazar.metadata[RDF.type] = [OT.Model, OTA.RegressionLazySingleTarget]
     end
 
     lazar.metadata[OT.parameters] = [