From 7e824ae1a52ee27bda90dd08783aef0ab3a539a9 Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Fri, 29 Jan 2016 12:23:14 +0100 Subject: internal calculation of difference features --- test/predict.rb | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) (limited to 'test/predict.rb') 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 -- cgit v1.2.3