1 files changed, 1 insertions, 4 deletions
diff --git a/lazar.rb b/lazar.rb
index a12fa18..61c2485 100644
--- a/lazar.rb
+++ b/lazar.rb
@@ -73,10 +73,7 @@ post '/lazar/?' do
     # Default Values
     # Classification: Weighted Majority, Substructure.match
     if prediction_feature.feature_type == "classification"
-      @training_classes = training_activities.accept_values(prediction_feature.uri).sort
-      @training_classes.each_with_index { |c,i|
-        lazar.value_map[i+1] = c # don't use '0': we must take the weighted mean later.
-      }
+      lazar.value_map = training_activities.value_map(params[:prediction_feature])
     # Regression: SVM, Substructure.match_hits
     elsif  prediction_feature.feature_type == "regression"
       lazar.feature_calculation_algorithm = "Substructure.match_hits"