diff options
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 |