internal calculation of difference featuresprediction

1 files changed, 8 insertions, 6 deletions

diff --git a/test/predict.rb b/test/predict.rb
index 9aed5e2..c72de02 100644
--- a/test/predict.rb
+++ b/test/predict.rb
@@ -1,9 +1,11 @@
 require_relative "../nanoparticles.rb"
 data = JSON.parse(File.read("./data.json"))
-relevant_features = JSON.parse(File.read("./relevant-features.json"))
-example = data[data.keys.sample]["physchem"].select{|f,v| relevant_features.keys.include? f}
-#data.collect
-puts predict(example)[:match].collect{|id,v| v["tox"]}.first
-puts predict(example)[:prediction]
-#puts predict(example)[:neighbors].size
+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