summaryrefslogtreecommitdiff
path: root/test/experiment.rb
blob: c465d7bd070c56e1afa6cb9f8fb830dfb8fc7f2b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
require_relative "setup.rb"

class ExperimentTest < MiniTest::Test

  def test_regression_experiment
    datasets = [
      "EPAFHM.csv",
      "FDA_v3b_Maximum_Recommended_Daily_Dose_mmol.csv",
      "LOAEL_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
    p experiment
    experiment.report
=end
    refute_empty experiment.crossvalidation_ids
  end
end