diff options
Diffstat (limited to 'lib/model.rb')
-rw-r--r-- | lib/model.rb | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/lib/model.rb b/lib/model.rb index aed789c..36011a0 100644 --- a/lib/model.rb +++ b/lib/model.rb @@ -192,15 +192,19 @@ module OpenTox end def self.from_csv_file file - p file metadata_file = file.sub(/csv$/,"json") - p metadata_file bad_request_error "No metadata file #{metadata_file}" unless File.exist? metadata_file prediction_model = self.new JSON.parse(File.read(metadata_file)) training_dataset = Dataset.from_csv_file file - # TODO classification - model = LazarRegression.create training_dataset - cv = RegressionCrossValidation.create model + model = nil + cv = nil + if training_dataset.features.first.nominal? + model = LazarFminerClassification.create training_dataset + cv = ClassificationCrossValidation.create model + elsif training_dataset.features.first.numeric? + model = LazarRegression.create training_dataset + cv = RegressionCrossValidation.create model + end prediction_model[:model_id] = model.id prediction_model[:crossvalidation_id] = cv.id prediction_model.save |