forked from lisskor/firefox-translations-training
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Bicleaner support + fixes (mozilla#13)
SacreBLEU is a regular importer now and evaluation is not limited to sacrebleu datasets. fixes Added bicleaner-ai and bicleaner filtering (one or another based on available pretrained language packs). fixes Added script to find all datasets based on language pair and importer type, ready to use in config fixes Fixed conda environment activation to be reproducible on GCP Other minor reproducibility fixes
- Loading branch information
Showing
35 changed files
with
500 additions
and
123 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,60 @@ | ||
| #!/bin/bash | ||
| ## | ||
| # Cleans corpus using bicleaner-ai or bicleaner | ||
| # | ||
| # Usage: | ||
| # bash bicleaner.sh corpus_prefix output_prefix | ||
| # | ||
|
|
||
| set -x | ||
| set -euo pipefail | ||
|
|
||
| echo "###### Bicleaner filtering" | ||
|
|
||
| test -v SRC | ||
| test -v TRG | ||
| test -v CLEAN_TOOLS | ||
| test -v BICLEANER_THRESHOLD | ||
|
|
||
| corpus_prefix=$1 | ||
| output_prefix=$2 | ||
|
|
||
| output_dir=$(dirname "${output_prefix}") | ||
| tmp_dir="${output_dir}/tmp" | ||
| mkdir -p "${tmp_dir}" | ||
|
|
||
| source "${WORKDIR}/pipeline/setup/activate-python.sh" | ||
|
|
||
| # bicleaner and bicleaner-ai have conflicting dependencies. installing on demand | ||
| if [ ! -e "${output_prefix}.${SRC}.gz" ]; then | ||
| if bash "${CLEAN_TOOLS}/download-bicleaner-pack.sh" "${tmp_dir}" "bicleaner-ai"; then | ||
| echo "### Using bicleaner-ai" | ||
| pip install bicleaner-ai==1.0.1 | ||
| cmd=bicleaner-ai-classify | ||
| elif bash "${CLEAN_TOOLS}/download-bicleaner-pack.sh" "${tmp_dir}" "bicleaner"; then | ||
| echo "### Using bicleaner" | ||
| pip install bicleaner==0.14 | ||
| cmd=bicleaner-classify | ||
| else | ||
| echo "### Bicleaner language pack is not supported, skipping." | ||
| cp "${corpus_prefix}.${SRC}.gz" "${output_prefix}.${SRC}.gz" | ||
| cp "${corpus_prefix}.${TRG}.gz" "${output_prefix}.${TRG}.gz" | ||
| exit 0 | ||
| fi | ||
| fi | ||
|
|
||
| echo "### Classifying and filtering" | ||
| test -s "${output_prefix}.${SRC}.gz" || test -s "${tmp_dir}/best.gz" || | ||
| paste <(pigz -dc "${corpus_prefix}.${SRC}.gz") <(pigz -dc "${corpus_prefix}.${TRG}.gz") | | ||
| ${cmd} --scol 1 --tcol 1 - - "${tmp_dir}"/*.yaml | | ||
| awk -v threshold=${BICLEANER_THRESHOLD} '{if ($3>threshold) {print $0}}' | | ||
| pigz >"${tmp_dir}/best.gz" | ||
|
|
||
| echo "### Writing output corpus" | ||
| test -s "${output_prefix}.${SRC}.gz" || pigz -dc "${tmp_dir}/best.gz" | cut -f1 | pigz >"${output_prefix}.${SRC}.gz" | ||
| test -s "${output_prefix}.${TRG}.gz" || pigz -dc "${tmp_dir}/best.gz" | cut -f2 | pigz >"${output_prefix}.${TRG}.gz" | ||
|
|
||
| echo "### Cleaning files" | ||
| rm -rf "${tmp_dir}" | ||
|
|
||
| echo "###### Done: Bicleaner filtering" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.