diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/predict.rb | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/test/predict.rb b/test/predict.rb new file mode 100644 index 0000000..c72de02 --- /dev/null +++ b/test/predict.rb @@ -0,0 +1,11 @@ +require_relative "../nanoparticles.rb" +data = JSON.parse(File.read("./data.json")) +query_features = JSON.parse(File.read("./query-features.json")) +key = data.keys.sample +p key +example = data[key]["physchem"].select{|f,v| query_features.include? f} +prediction = predict(example) +puts prediction[:match].collect{|id,v| v["tox"]}.first +puts prediction[:prediction] +puts prediction[:neighbors].size + |