floret_word_embedding_train_save_load_quickstart.py

1import floret
2
3# train a model
4# for more details on parameters, see fasttext documentation
5model = floret.train_unsupervised(
6    "data.txt",
7    model="skipgram",
8    hashCount=2,
9    minn=3,
10    maxn=6,
11    dim=50,
12)
13
14# save the model
15model.save_model("model.bin")
16
17# load the model
18model = floret.load_model("model.bin")
19
20# check a vector
21print(model.get_word_vector("word"))