From bdc6b5b40437896384561d74a510560e9e592364 Mon Sep 17 00:00:00 2001 From: "helma@in-silico.ch" Date: Tue, 9 Oct 2018 18:20:27 +0200 Subject: tentative random forest classification: hangs unpredictably during caret model generation/optimization for some (inorganic?) compounds. --- test/dataset.rb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'test/dataset.rb') diff --git a/test/dataset.rb b/test/dataset.rb index 11b8d49..4196fd8 100644 --- a/test/dataset.rb +++ b/test/dataset.rb @@ -66,6 +66,7 @@ class DatasetTest < MiniTest::Test assert_equal Dataset, d.class assert_equal 1, d.features.size assert_equal 85, d.compounds.size + assert_equal true, d.features.first.measured csv = CSV.read("#{DATA_DIR}/hamster_carcinogenicity.csv") csv.shift csv.each do |row| @@ -174,7 +175,7 @@ class DatasetTest < MiniTest::Test hansen = Dataset.from_csv_file "#{DATA_DIR}/hansen.csv" efsa = Dataset.from_csv_file "#{DATA_DIR}/efsa.csv" d = Dataset.merge [kazius,hansen,efsa], {source_feature => target_feature}, {1 => "mutagen", 0 => "nonmutagen"} - File.open("tmp.csv","w+"){|f| f.puts d.to_csv} + #File.open("tmp.csv","w+"){|f| f.puts d.to_csv} assert_equal 8281, d.compounds.size assert_equal 4, d.features.size c = Compound.from_smiles("C/C=C/C=O") -- cgit v1.2.3