made y.argmax hdf compatible
@@ -514,7 +514,7 @@ def _standardize_weights(y, sample_weight=None, class_weight=None,
raise ValueError('`class_weight` not supported for '
'3+ dimensional targets.')
if y.shape[1] > 1:
- y_classes = y.argmax(axis=1)
+ y_classes = np.argmax(y, axis=1)
elif y.shape[1] == 1:
y_classes = np.reshape(y, y.shape[0])
else:
