Commit 0a03b138 authored by leswing's avatar leswing
Browse files

to_one_hot safe

parent 90d01863
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -14,7 +14,7 @@ from sklearn.metrics import precision_score
from scipy.stats import pearsonr


def to_one_hot(y, axis=-1, n_classes=2):
def to_one_hot(y, n_classes=2):
  """Transforms label vector into one-hot encoding.

  Turns y into vector of shape [n_samples, 2] (assuming binary labels).