summaryrefslogtreecommitdiff
path: root/test/classification-model.rb
diff options
context:
space:
mode:
authorhelma@in-silico.ch <helma@in-silico.ch>2018-10-29 17:58:09 +0100
committerhelma@in-silico.ch <helma@in-silico.ch>2018-10-29 17:58:09 +0100
commit5e9a08c0b534fa96179fb5c81a9b4193e7b0aad8 (patch)
treea9d97dc520f51b5bca61b5f0122a1f86817e1ea8 /test/classification-model.rb
parent15f4ad23eb918a91d52779887ccfb51bc6547f1b (diff)
dataset folds fixed
Diffstat (limited to 'test/classification-model.rb')
-rw-r--r--test/classification-model.rb1
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