Skip to content

Commit

Permalink
better test for transposing
Browse files Browse the repository at this point in the history
  • Loading branch information
lbulgarelli committed Nov 19, 2019
1 parent d99328e commit f64c440
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pyroc.py
Original file line number Diff line number Diff line change
Expand Up @@ -114,7 +114,7 @@ def _parse_inputs(self, preds, target):
parsed = np.array([preds])

# If needed to transpose matrix
if parsed.shape[1] != len(target):
if parsed.shape[0] == len(target):
parsed = preds.T

# Use column names if it is a DataFrame
Expand Down

0 comments on commit f64c440

Please sign in to comment.