EDSL code
ME-98K dataset: https://pan.baidu.com/s/1itQEjPUMve3A3dXezJDedw password: axqh
ME-20K dataset: https://pan.baidu.com/s/1ti1xfCdj6c36yvy_sHld5Q password: nwks
Preprocessing of training set.
python scrips/preprocessing/preprocess_train.py --formulas data/sample/formulas.txt --train data/sample/train.txt --vocab data/sample/latex_vocab.txt --img data/sample/image_processed/
Preprocessing of validation set
python scrips/preprocessing/preprocess_val.py --formulas data/sample/formulas.txt --val data/sample/train.txt --vocab data/sample/latex_vocab.txt --img data/sample/image_processed/
Preprocessing of test set
python scrips/preprocessing/preprocess_test.py --formulas data/sample/formulas.txt --test data/sample/test.txt --vocab data/sample/latex_vocab.txt --img data/sample/image_processed/
python src/train.py --formulas data/sample/formulas.txt --train data/sample/train.txt --val data/sample/val.txt --vocab data/sample/latex_vocab.txt
python src/train.py --formulas data/sample/formulas.txt --test data/sample/test.txt --vocab data/sample/latex_vocab.txt
BLEU-4 Calculation:
python scrips/evaluation/Cal_B4.py --formulas data/sample/formulas.txt
Rouge-4 Calculation:
python scrips/evaluation/Cal_R4.py --formulas data/sample/formulas.txt
Match Calculation:
python scrips/evaluation/CalculateMath.py --formulas data/sample/formulas.txt
Match-ws Calculation:
python scrips/evaluation/CalculateMath-WS.py --formulas data/sample/formulas.txt