From 65f7f07c23606becea4ee90fd8047dc52a2b28c0 Mon Sep 17 00:00:00 2001 From: vandanavk Date: Tue, 7 Aug 2018 07:58:43 -0700 Subject: [PATCH] Fix for undefined names in examples --- example/rnn/word_lm/train.py | 2 +- example/sparse/factorization_machine/train.py | 2 +- example/sparse/wide_deep/train.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/example/rnn/word_lm/train.py b/example/rnn/word_lm/train.py index aa641358cf90..c48fe800ce9a 100644 --- 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 index c73ce88d41c2..b30f9cc81acf 100644 --- 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 index 89befb5aa82d..6fd81b7fa480 100644 --- 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 * +from data import get_uci_adult +from model import wide_deep_model import argparse import os