From a10fabfc533b4f21eff0fc102ced5b1d2c3e3699 Mon Sep 17 00:00:00 2001 From: mguetlein Date: Mon, 6 Oct 2014 21:44:59 +0200 Subject: fix split method for datasets with missing values --- lib/dataset.rb | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'lib') diff --git a/lib/dataset.rb b/lib/dataset.rb index 292b26b..83ef12b 100644 --- a/lib/dataset.rb +++ b/lib/dataset.rb @@ -308,7 +308,11 @@ module OpenTox dataset.metadata = metadata dataset.features = (feats ? feats : self.features) compound_indices.each do |c_idx| - dataset << [ self.compounds[c_idx] ] + dataset.features.each_with_index.collect{|f,f_idx| self.data_entries[c_idx][f_idx]} + d = [ self.compounds[c_idx] ] + dataset.features.each_with_index.each do |f,f_idx| + d << (self.data_entries[c_idx] ? self.data_entries[c_idx][f_idx] : nil) + end + dataset << d end dataset.put dataset -- cgit v1.2.3