Continue training a FastText model

Viewed 5910

I have downloaded a .bin FastText model, and I use it with gensim as follows:

model = FastText.load_fasttext_format("cc.fr.300.bin")

I would like to continue the training of the model to adapt it to my domain. After checking FastText's Github and the Gensim documentation it seems like it is not currently feasible appart from using this person's proposed modification (not yet merged).

Am I missing something?

2 Answers

You can continue training in some versions of Gensim's fastText (for example, v.3.7.*). Here is an example of "Loading, inferring, continuing training"

from gensim.test.utils import datapath
model = load_facebook_model(datapath("crime-and-punishment.bin"))
sent = [['lord', 'of', 'the', 'rings'], ['lord', 'of', 'the', 'semi-groups']]
model.build_vocab(sent, update=True)
model.train(sentences=sent, total_examples = len(sent), epochs=5)

For some reason, the gensim.models.fasttext.load_facebook_model() is missing on Windows, but exists on Mac's installation. Alternatively, one can use gensim.models.FastText.load_fasttext_format() to load a pre-trained model and continue training.

Here are various pre-trained Wiki word models and vectors (or here).

Another example. "Note: As in the case of Word2Vec, you can continue to train your model while using Gensim's native implementation of fastText."

The official FastText implementation currently doesn't support that, although there is an open ticket related to this issue which you can find here.

Related