guochaorong-patch-1
weixing02 7 years ago
parent 34d92eaddf
commit 2f49432538

@ -3028,9 +3028,9 @@ def hsigmoid(input, label, num_classes=2, param_attr=None, bias_attr=None):
pre_out = helper.create_tmp_variable(dtype)
dim = input.shape[1]
if num_classes < 2:
raise valueError("num_classes must be lager or equal than 2.")
if x.shape[0] != y.shape[1]:
raise valueError(
raise ValueError("num_classes must be lager or equal than 2.")
if input.shape[0] != label.shape[1]:
raise ValueError(
"input's 1-st dimension and label's 2-nd dimension must be equal they both equal to batch size."
)
weights = helper.create_parameter(

Loading…
Cancel
Save