| #!/bin/bash |
| |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| |
| MXNET_ROOT=$(cd "$(dirname $0)/../../.."; pwd) |
| CLASS_PATH=$MXNET_ROOT/scala-package/assembly/target/*:$MXNET_ROOT/scala-package/examples/target/*:$MXNET_ROOT/scala-package/examples/target/classes/lib/* |
| |
| # which gpu card to use, -1 means cpu |
| GPU=$1 |
| # the mr dataset path, you should put the pos and neg file in the same folder |
| MR_DATASET_PATH=$2 |
| # the trained word2vec file path, binary or text format |
| W2V_FILE_PATH=$3 |
| # whether the format of the word2vec file is binary,1 means binary, 0 means text |
| W2V_FORMAT_BIN=$4 |
| BATCH_SIZE=$5 |
| SAVE_MODEL_PATH=$6 |
| |
| java -Xmx8G -cp $CLASS_PATH \ |
| org.apache.mxnetexamples.cnntextclassification.CNNTextClassification \ |
| --gpu $GPU \ |
| --mr-dataset-path $MR_DATASET_PATH \ |
| --w2v-file-path $W2V_FILE_PATH \ |
| --w2v-format-bin $W2V_FORMAT_BIN \ |
| --batch-size $BATCH_SIZE \ |
| --save-model-path $SAVE_MODEL_PATH |
