summaryrefslogtreecommitdiff
path: root/scripts/test-validation.rb
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/test-validation.rb')
-rwxr-xr-xscripts/test-validation.rb2
1 files changed, 1 insertions, 1 deletions
diff --git a/scripts/test-validation.rb b/scripts/test-validation.rb
index b64edd6..0b8c0a7 100755
--- a/scripts/test-validation.rb
+++ b/scripts/test-validation.rb
@@ -5,6 +5,6 @@ include OpenTox
test = Dataset.from_csv_file(File.join("data","test_log10.csv"))
train = Dataset.from_csv_file(File.join("data","training_log10.csv"))
-model = Model::LazarRegression.create(training_dataset: train, algorithms: { :similarity => { :min => 0.5 }})
+model = Model::LazarRegression.create(training_dataset: train, algorithms: { :prediction => {:method => "Algorithm::Caret.rf"}, :similarity => { :min => 0.5 }})
validation = Validation::TrainTest.create model, train, test
File.open(File.join("data","training-test-predictions.id"),"w+") { |f| f.puts validation.id }