fix: retrain for real

dhdaines · Jul 19, 2024 · f841e70 · f841e70
1 parent f090aeb
commit f841e70
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 0 deletions.
diff --git a/alexi/models/crf.joblib.gz b/alexi/models/crf.joblib.gz
diff --git a/alexi/models/crf.vl.joblib.gz b/alexi/models/crf.vl.joblib.gz
diff --git a/alexi/models/crfseq.joblib.gz b/alexi/models/crfseq.joblib.gz
diff --git a/alexi/models/rnn.pt b/alexi/models/rnn.pt
diff --git a/alexi/models/rnn_crf.pt b/alexi/models/rnn_crf.pt
diff --git 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 \\