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