Experiments added

1 files changed, 31 insertions, 0 deletions

diff --git a/test/experiment.rb b/test/experiment.rb
new file mode 100644
index 0000000..eae7fa0
--- /dev/null
+++ b/test/experiment.rb
@@ -0,0 +1,31 @@
+require_relative "setup.rb"
+
+class ExperimentTest < MiniTest::Test
+
+  def test_regression_experiment
+    datasets = [
+      "EPAFHM.csv",
+      "FDA_v3b_Maximum_Recommended_Daily_Dose_mmol.csv",
+      "LOAEL_log_mmol_corrected_smiles.csv"
+      ]
+    model_algorithms = ["OpenTox::Model::LazarRegression"]
+    neighbor_algorithms = ["OpenTox::Algorithm::Neighbor.fingerprint_similarity"]
+    prediction_algorithms = ["OpenTox::Algorithm::Regression.weighted_average"]
+    neighbor_algorithm_parameters = [{:min_sim => 0.7}]
+    experiment = Experiment.create(
+      :name => "Regression for datasets #{datasets}.",
+      :dataset_ids => datasets.collect{|d| Dataset.from_csv_file(File.join(DATA_DIR, d)).id},
+      :model_algorithms => model_algorithms,
+      :neighbor_algorithms => neighbor_algorithms,
+      :neighbor_algorithm_parameters => neighbor_algorithm_parameters,
+      :prediction_algorithms => prediction_algorithms,
+    )
+    experiment.run
+=begin
+    experiment = Experiment.find "55dc58b32b72ed14a8000008"
+=end
+    p experiment.id
+    experiment.report
+    refute_empty experiment.crossvalidation_ids
+  end
+end