Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Make api call consistent
Browse files Browse the repository at this point in the history
  • Loading branch information
Ubuntu committed May 1, 2018
1 parent 8727cae commit ad1dc3d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions example/sparse/factorization_machine/model.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -27,10 +27,10 @@ def factorization_machine_model(factor_size, num_features,
v = mx.symbol.Variable("v", shape=(num_features, factor_size), stype='row_sparse',
init=init_config['v'], lr_mult=lr_mult_config['v'],
wd_mult=wd_mult_config['v'])
w = mx.symbol.var('w', shape=(num_features, 1), stype='row_sparse',
w = mx.symbol.Variable('w', shape=(num_features, 1), stype='row_sparse',
init=init_config['w'], lr_mult=lr_mult_config['w'],
wd_mult=wd_mult_config['w'])
w0 = mx.symbol.var('w0', shape=(1,), init=init_config['w0'],
w0 = mx.symbol.Varialble('w0', shape=(1,), init=init_config['w0'],
lr_mult=lr_mult_config['w0'], wd_mult=wd_mult_config['w0'])
w1 = mx.symbol.broadcast_add(mx.symbol.dot(x, w), w0)

Expand Down

1 comment on commit ad1dc3d

@jamesliu
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I feel it is confusing to use API in a not consistent way so I request to make it consistent.

Thanks,
James

Please sign in to comment.