module SecEdgar::BasicStats

Public Instance Methods

f1(tp, fp, fn) click to toggle source
# File lib/sec_edgar/basic_stats.rb, line 20
def f1(tp, fp, fn)
  p = precision(tp, fp)
  r = recall(tp, fn)
  2 * ((p * r).fdiv(p + r))
end
f_beta(tp, fp, fn, beta) click to toggle source
# File lib/sec_edgar/basic_stats.rb, line 13
def f_beta(tp, fp, fn, beta)
  p = precision(tp, fp)
  r = recall(tp, fn)
  b2 = beta ** 2
  (1 + b2) * ((p * r).fdiv((b2 * p) + r))
end
precision(tp, fp) click to toggle source
# File lib/sec_edgar/basic_stats.rb, line 5
def precision(tp, fp)
  tp.fdiv(tp + fp)
end
recall(tp, fn) click to toggle source
# File lib/sec_edgar/basic_stats.rb, line 9
def recall(tp, fn)
  tp.fdiv(tp + fn)
end