From e5c538f6c71e7c17c63d7233ef1be06a19b9e132 Mon Sep 17 00:00:00 2001
From: Christoph Helma <helma@in-silico.ch>
Date: Mon, 14 Sep 2015 11:44:34 +0200
Subject: classification? and regression? methods for Model::Prediction

---
 lib/model.rb | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/lib/model.rb b/lib/model.rb
index 547144f..d89051a 100644
--- a/lib/model.rb
+++ b/lib/model.rb
@@ -194,6 +194,14 @@ module OpenTox
         CrossValidation.find crossvalidation_id
       end
 
+      def regression?
+        training_dataset.features.first.numeric?
+      end
+
+      def classification?
+        training_dataset.features.first.nominal?
+      end
+
       def self.from_csv_file file
         metadata_file = file.sub(/csv$/,"json")
         bad_request_error "No metadata file #{metadata_file}" unless File.exist? metadata_file
-- 
cgit v1.2.3