Commit b1c67590 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

Consider non 1 values negatives

parent 751d030e
Pipeline #21390 passed with stage
in 36 minutes and 14 seconds
...@@ -43,8 +43,8 @@ def split(filename): ...@@ -43,8 +43,8 @@ def split(filename):
positive) and the second the scores positive) and the second the scores
(float).'''.format(filename)) (float).'''.format(filename))
raise raise
return (scores[numpy.where(neg_pos == -1)], positives = neg_pos == 1
scores[numpy.where(neg_pos == 1)]) return (scores[~positives], scores[positives])
def split_files(filenames): def split_files(filenames):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment