aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorCalvin Morrison <mutantturkey@gmail.com>2014-03-25 17:48:31 -0400
committerCalvin Morrison <mutantturkey@gmail.com>2014-03-25 17:48:31 -0400
commita62572ad5ccbc1e968a5164daaaf690e2881062d (patch)
tree8144c789c2a7b221afdde3074e746f95ed0b76e7 /src
parentc0afca51eeebd5a7654e2350ebba5d2e46e6bc18 (diff)
update selectivity equation, and update readme
Diffstat (limited to 'src')
-rwxr-xr-xsrc/select_mers.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/select_mers.py b/src/select_mers.py
index 5bd6877..3a24ad4 100755
--- a/src/select_mers.py
+++ b/src/select_mers.py
@@ -5,6 +5,8 @@ import os
fg_mers = {}
bg_mers = {}
+fg_weight = int(os.environ.get("fg_weight", 0))
+
if(len(sys.argv) == 3):
fg_count_fn = sys.argv[1]
bg_count_fn = sys.argv[2]
@@ -32,7 +34,7 @@ def select_mers(fg_mers, bg_mers):
fg_arr = np.array(fg_arr, dtype='f');
bg_arr = np.array(bg_arr, dtype='f');
- selectivity = (fg_arr / bg_arr)
+ selectivity = (fg_arr / bg_arr) * (fg_arr**fg_weight)
arr = [(mers[i], fg_arr[i], bg_arr[i], selectivity[i]) for i in range(len(mers))]