update

jmgx30 · Jun 12, 2019 · bcd8d68 · bcd8d68
1 parent ed2c6c6
commit bcd8d68
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/codes/data_process.py b/codes/data_process.py
@@ -34,11 +34,10 @@ def write_file(folder,fin):
     print(len(lines))
     p = progressbar.ProgressBar()
     for i in p(range(len(lines))):  
-        line = lines[i]
+        line = lines[i].strip()
         strs = line.split("\t");
         reaction = strs[0]
-        label = '1'
-        if strs[1]=='0': label = '0'
+        label = strs[1]
         strss = reaction.split(">")
         if not len(strss)==3: continue
         line_l = SplitByElement.split_line(strss[0]);