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):
positive) and the second the scores
(float).'''.format(filename))
raise
return (scores[numpy.where(neg_pos == -1)],
scores[numpy.where(neg_pos == 1)])
positives = neg_pos == 1
return (scores[~positives], scores[positives])
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