diff options
author | helma@in-silico.ch <helma@in-silico.ch> | 2018-10-29 17:58:09 +0100 |
---|---|---|
committer | helma@in-silico.ch <helma@in-silico.ch> | 2018-10-29 17:58:09 +0100 |
commit | 5e9a08c0b534fa96179fb5c81a9b4193e7b0aad8 (patch) | |
tree | a9d97dc520f51b5bca61b5f0122a1f86817e1ea8 /test/classification-model.rb | |
parent | 15f4ad23eb918a91d52779887ccfb51bc6547f1b (diff) |
dataset folds fixed
Diffstat (limited to 'test/classification-model.rb')
-rw-r--r-- | test/classification-model.rb | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/test/classification-model.rb b/test/classification-model.rb index 7a2a64f..bfb64db 100644 --- a/test/classification-model.rb +++ b/test/classification-model.rb @@ -84,6 +84,7 @@ class LazarClassificationTest < MiniTest::Test training_dataset = Dataset.from_csv_file File.join(DATA_DIR,"hamster_carcinogenicity.csv") model = Model::Lazar.create training_dataset: training_dataset result = model.predict training_dataset + puts result.to_csv assert_kind_of Dataset, result assert 3, result.features.size assert 8, result.compounds.size |