classifs / tags /train.py
anshy's picture
Upload folder using huggingface_hub
7f390fb verified
Raw History Blame Contribute Delete
580 Bytes
"""Exact training call used for this model. Run from inside this directory:
python train.py
Same dim64/epoch10 baseline hyperparams as quality/multilingual, plus loss="ova"
(fastText's native multi-label mode): one shared model/vocabulary across all 424 tags,
trained jointly, rather than 424 separate binary classifiers -- see README.md."""
import fasttext
HPARAMS = dict(dim=64, epoch=10, lr=0.3, wordNgrams=2, minCount=3, bucket=2_000_000, loss="ova")
model = fasttext.train_supervised(input="train.txt", **HPARAMS)
model.save_model("model.bin")
print("saved model.bin")