summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/model.rb8
1 files changed, 8 insertions, 0 deletions
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