update scripts and readme

supercoderhawk · Jan 16, 2020 · 9a801de · 9a801de
1 parent 7e5ed7f
commit 9a801de
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/ReadMe.md b/ReadMe.md
@@ -24,9 +24,11 @@ python3 -m spacy download en
 1. unzip file and put all files under `data/` folder, rename `test.csv` to `test_release.csv`
 2. execute `bash scripts/prepare_data.sh` in **project root folder** to build the data for next step
 
-### Execute the retrieval process
+### Execute the retrieval process end2end
 
-execute `bash scripts/run_retrieval.sh` in **project root folder**
+* put the model into `data/models/rerank_model.model`
+
+* execute `bash scripts/run_end2end.sh` in **project root folder**
 
 #### details
 the above script includes three main parts

diff --git a/scripts/run_end2end.sh b/scripts/run_end2end.sh
@@ -3,7 +3,7 @@
 DATA_DIR=${PWD}/data/
 ES_RESULT_FILE=$DATA_DIR/validation_es_result.jsonl
 FINAL_RESULT_FILENAME=$DATA_DIR/validation_final_result.jsonl
-MODEL_PATH=$MODEL_DIR/rerank_model.model
+MODEL_PATH=$DATA_DIR/models/rerank_model.model
 TOPK=20
 
 # run elasticsearch (BM25)