API calls are not consistent in the example (#10763)

* Make api call consistent * Make API call consistent
apache · May 1, 2018 · 97da5e3 · 97da5e3
1 parent 1f42f2c
commit 97da5e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/example/sparse/factorization_machine/model.py b/example/sparse/factorization_machine/model.py
@@ -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.Variable('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)