Fix parser model for depth 1

This commit is contained in:
Matthw Honnibal 2019-10-23 05:14:00 +02:00
parent 9e32d8271c
commit 9ca109597d

View File

@ -48,8 +48,8 @@ cdef WeightsC get_c_weights(model) except *:
output.hidden_weights = NULL output.hidden_weights = NULL
output.hidden_bias = NULL output.hidden_bias = NULL
else: else:
vec2scores_W = model.vec2scores.b vec2scores_W = model.vec2scores.W
vec2scores_b = model.vec2scores.W vec2scores_b = model.vec2scores.b
output.hidden_weights = <const float*>vec2scores_W.data output.hidden_weights = <const float*>vec2scores_W.data
output.hidden_bias = <const float*>vec2scores_b.data output.hidden_bias = <const float*>vec2scores_b.data
cdef np.ndarray class_mask = model._class_mask cdef np.ndarray class_mask = model._class_mask