feature selection added

1 files changed, 9 insertions, 0 deletions

diff --git a/test/predict.rb b/test/predict.rb
new file mode 100644
index 0000000..9aed5e2
--- /dev/null
+++ b/test/predict.rb
@@ -0,0 +1,9 @@
+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
+