(self, model_file)
| 408 | self._load_fasttext_model(model_file) |
| 409 | |
| 410 | def _load_fasttext_model(self, model_file): |
| 411 | from fastText import load_model |
| 412 | from pythia.common.registry import registry |
| 413 | |
| 414 | pythia_root = get_pythia_root() |
| 415 | model_file = os.path.join(pythia_root, model_file) |
| 416 | |
| 417 | registry.get("writer").write("Loading fasttext model now from %s" % model_file) |
| 418 | |
| 419 | self.model = load_model(model_file) |
| 420 | self.stoi = WordToVectorDict(self.model) |
| 421 | |
| 422 | def get_embedding_dim(self): |
| 423 | return self.model.get_dimension() |
no test coverage detected