diff options
Diffstat (limited to 'test/predict.rb')
-rw-r--r-- | test/predict.rb | 9 |
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 + |