gen_model.py

# %%
from game import Semantrix, Model_class

# %%
model = "word2vec"
model_class = Model_class(lang=0, model_type=model)

sem = Semantrix(lang=1, model_type="SentenceTransformer")

# %%
sem.generate_gensim_model(batch_size=128, model_class=model_class)

# %%