Fix for undefined names in examples

apache · Aug 7, 2018 · 9268120 · 9268120
1 parent 3a5b97e
commit 9268120
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/example/rnn/word_lm/train.py b/example/rnn/word_lm/train.py
@@ -20,7 +20,7 @@
 import argparse, math
 import logging
 from data import Corpus, CorpusIter
-from model import *
+from model import rnn, softmax_ce_loss
 from module import *
 from mxnet.model import BatchEndParam
 

diff --git a/example/sparse/factorization_machine/train.py b/example/sparse/factorization_machine/train.py
@@ -18,7 +18,7 @@
 import mxnet as mx
 from metric import *
 from mxnet.test_utils import *
-from model import *
+from model import factorization_machine_model
 import argparse, os
 
 parser = argparse.ArgumentParser(description="Run factorization machine with criteo dataset",

diff --git a/example/sparse/wide_deep/train.py b/example/sparse/wide_deep/train.py
@@ -17,8 +17,8 @@
 
 import mxnet as mx
 from mxnet.test_utils import *
-from data import *
-from model import *
+import data
+from model import wide_deep_model
 import argparse
 import os
 
@@ -72,8 +72,8 @@
     data_dir = os.path.join(os.getcwd(), 'data')
     train_data = os.path.join(data_dir, ADULT['train'])
     val_data = os.path.join(data_dir, ADULT['test'])
-    train_csr, train_dns, train_label = get_uci_adult(data_dir, ADULT['train'], ADULT['url'])
-    val_csr, val_dns, val_label = get_uci_adult(data_dir, ADULT['test'], ADULT['url'])
+    train_csr, train_dns, train_label = data.get_uci_adult(data_dir, ADULT['train'], ADULT['url'])
+    val_csr, val_dns, val_label = data.get_uci_adult(data_dir, ADULT['test'], ADULT['url'])
 
     model = wide_deep_model(ADULT['num_linear_features'], ADULT['num_embed_features'],
                             ADULT['num_cont_features'], ADULT['embed_input_dims'],