This is an automated email from the ASF dual-hosted git repository. cmeier pushed a commit to branch clojure-bert-qa-example in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 9a15c06d52636cd96a8a3b768dfae6e5527cf043 Author: gigasquid <[email protected]> AuthorDate: Fri Apr 12 17:17:15 2019 -0400 prediction --- .../examples/bert-qa/src/bert_qa/core.clj | 26 +++++++++++++++++----- 1 file changed, 21 insertions(+), 5 deletions(-) diff --git a/contrib/clojure-package/examples/bert-qa/src/bert_qa/core.clj b/contrib/clojure-package/examples/bert-qa/src/bert_qa/core.clj index 941e6aa..1876a83 100644 --- a/contrib/clojure-package/examples/bert-qa/src/bert_qa/core.clj +++ b/contrib/clojure-package/examples/bert-qa/src/bert_qa/core.clj @@ -7,7 +7,8 @@ [org.apache.clojure-mxnet.dtype :as dtype] [org.apache.clojure-mxnet.context :as context] [org.apache.clojure-mxnet.layout :as layout] - [org.apache.clojure-mxnet.ndarray :as ndarray]) + [org.apache.clojure-mxnet.ndarray :as ndarray] + [org.apache.clojure-mxnet.infer :as infer]) (:import (bert BertDataParser) (bert BertQA))) @@ -81,10 +82,25 @@ input-batch [(ndarray/array indexes [1 seq-length] {:context ctx}) (ndarray/array token-types [1 seq-length] {:context ctx}) (ndarray/array [valid-length] [1] {:context ctx})] - - - ] - input-batch) + input-descs [{:name "data0" + :shape [1 seq-length] + :dtype dtype/FLOAT32 + :layout layout/NT} + {:name "data1" + :shape [1 seq-length] + :dtype dtype/FLOAT32 + :layout layout/NT} + {:name "data2" + :shape [1] + :dtype dtype/FLOAT32 + :layout layout/N}] + factory (infer/model-factory model-path-prefix input-descs) + predictor (infer/create-predictor + factory + {:contexts [ctx] + :epoch 2}) + result (first (infer/predict-with-ndarray predictor input-batch))] + result) )
