diff options
| author | helma@in-silico.ch <helma@in-silico.ch> | 2018-10-09 18:20:27 +0200 |
|---|---|---|
| committer | helma@in-silico.ch <helma@in-silico.ch> | 2018-10-09 18:20:27 +0200 |
| commit | bdc6b5b40437896384561d74a510560e9e592364 (patch) | |
| tree | a77e74803bc06157ac42c722f95884c2da163a75 /test/dataset.rb | |
| parent | 0a8da103e020b4a584a28a52b4ba12e1f3f90fd3 (diff) | |
tentative random forest classification: hangs unpredictably during caret model generation/optimization for some (inorganic?) compounds.mutanew
Diffstat (limited to 'test/dataset.rb')
| -rw-r--r-- | test/dataset.rb | 3 |
1 files changed, 2 insertions, 1 deletions
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") |