text segmentation
1.0.0
該存儲庫包含代碼和補充材料,這些材料需要培訓和評估紙質文本細分中所述的模型作為監督的學習任務
Wiki-727K,Wiki-50數據集:
https://www.dropbox.com/sh/k3jh0fjbyr0gw0a/aadzad9sdtrbnvs1qlcjy5cza?dl=0
Word2Vec:
https://drive.google.com/a/audioburst.com/uc?export=Download&confirm=zrin&id=0b7xkcwpi5kdynlnuttlss21pqmm
在configgenerator.py中填寫相關路徑,然後執行腳本(git存儲庫包括choi dataset)
conda create -n textseg python=2.7 numpy scipy gensim ipython
source activate textseg
pip install http://download.pytorch.org/whl/cu80/torch-0.3.0-cp27-cp27mu-linux_x86_64.whl
pip install tqdm pathlib2 segeval tensorboard_logger flask flask_wtf nltk
pip install pandas xlrd xlsxwriter termcolor
python run.py --help
例子:
python run.py --cuda --model max_sentence_embedding --wiki
python test_accuracy.py --help
例子:
python test_accuracy.py --cuda --model <path_to_model> --wiki
python wiki_processor.py --input <input> --temp <temp_files_folder> --output <output_folder> --train <ratio> --test <ratio>
輸入是通往Wikipedia轉儲的完整路徑,溫度是通往臨時文件文件夾的路徑,輸出是新生成的Wikipedia數據集的路徑。
可以從以下URL下載Wikipedia轉儲:
https://dumps.wikimedia.org/enwiki/latest/enwiki-latest-pages-articles.xml.bz2