summaryrefslogtreecommitdiff
path: root/test/predict.rb
blob: 9aed5e26dd7b70488028c7d722881a6ce5be75a3 (plain)
1
2
3
4
5
6
7
8
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