summaryrefslogtreecommitdiff
path: root/paper/test-validation.rb
diff options
context:
space:
mode:
Diffstat (limited to 'paper/test-validation.rb')
-rw-r--r--paper/test-validation.rb6
1 files changed, 4 insertions, 2 deletions
diff --git a/paper/test-validation.rb b/paper/test-validation.rb
index b748a8d..5edd6e6 100644
--- a/paper/test-validation.rb
+++ b/paper/test-validation.rb
@@ -1,10 +1,12 @@
require_relative "include.rb"
-test = Dataset.from_csv_file(File.join(DATA,"common-test.csv"))
+test = Dataset.from_csv_file(File.join(DATA,"test.csv"))
file = File.join(DATA,ARGV[0])
dataset = Dataset.from_csv_file file
-model = Model::LazarRegression.create dataset
+model = Model::LazarRegression.create(dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_physchem_regression")
+#model = Model::LazarRegression.create(dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_fingerprint_regression")
+#model = Model::LazarRegression.create(dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_weighted_average")
validation = RegressionValidation.create model, dataset, test
csv_file = file.sub(".csv","-test-predictions.csv")
id_file = file.sub(".csv","-test-predictions.id")