diff options
-rw-r--r-- | lib/nlp_ruby/SparseVector.rb | 15 | ||||
-rw-r--r-- | lib/nlp_ruby/misc.rb | 8 |
2 files changed, 22 insertions, 1 deletions
diff --git a/lib/nlp_ruby/SparseVector.rb b/lib/nlp_ruby/SparseVector.rb index cdf966c..1c0262b 100644 --- a/lib/nlp_ruby/SparseVector.rb +++ b/lib/nlp_ruby/SparseVector.rb @@ -1,8 +1,11 @@ class SparseVector < Hash - def initialize + def initialize arg=nil super self.default = 0 + if arg.is_a? Array + from_a arg + end end def from_a a @@ -21,6 +24,16 @@ class SparseVector < Hash self.values.inject(:+) end + def approx_eql? other, p=10**-10 + return false if !other + return false if other.size!=self.size + return false if other.keys.sort!=self.keys.sort + self.keys.each { |k| + return false if (self[k]-other[k]).abs>p + } + return true + end + def average self.sum/self.size.to_f end diff --git a/lib/nlp_ruby/misc.rb b/lib/nlp_ruby/misc.rb index 1fa3878..80d932c 100644 --- a/lib/nlp_ruby/misc.rb +++ b/lib/nlp_ruby/misc.rb @@ -11,6 +11,14 @@ class Array } return true end + + def sum + self.inject(:+) + end + + def mean + self.sum.to_f/self.size + end end def spawn_with_timeout cmd, t=4, debug=false |