diff --git a/alexi/models/crf.joblib.gz b/alexi/models/crf.joblib.gz index e01f58f..57a0bb1 100644 Binary files a/alexi/models/crf.joblib.gz and b/alexi/models/crf.joblib.gz differ diff --git a/alexi/models/crf.vl.joblib.gz b/alexi/models/crf.vl.joblib.gz index 1d19276..bb3235f 100644 Binary files a/alexi/models/crf.vl.joblib.gz and b/alexi/models/crf.vl.joblib.gz differ diff --git a/alexi/models/crfseq.joblib.gz b/alexi/models/crfseq.joblib.gz index df19840..531b5cd 100644 Binary files a/alexi/models/crfseq.joblib.gz and b/alexi/models/crfseq.joblib.gz differ diff --git a/alexi/models/rnn.pt b/alexi/models/rnn.pt index 8b9cc2b..2411096 100644 Binary files a/alexi/models/rnn.pt and b/alexi/models/rnn.pt differ diff --git a/alexi/models/rnn_crf.pt b/alexi/models/rnn_crf.pt index 0d4a26e..0484dfb 100644 Binary files a/alexi/models/rnn_crf.pt and b/alexi/models/rnn_crf.pt differ diff --git a/pyproject.toml b/pyproject.toml index ac01107..9c6ef9c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -76,6 +76,10 @@ train = [ -o alexi/models/rnn_crf.pt data/*.csv data/patches/*.csv""", """python scripts/train_crf.py \\ --features text+layout+structure --labels bonly \\ + --outfile alexi/models/crf.joblib.gz \\ + data/*.csv data/patches/*.csv""", +"""python scripts/train_crf.py \\ + --features text+layout --labels bonly \\ --outfile alexi/models/crf.vl.joblib.gz \\ data/*.csv data/patches/*.csv""", """python scripts/train_crf_seq.py \\