summaryrefslogtreecommitdiff
path: root/test/predict.rb
diff options
context:
space:
mode:
Diffstat (limited to 'test/predict.rb')
-rw-r--r--test/predict.rb11
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
+