def test_get_fasttext_model(self): data = pandas.read_csv(str(TEST_DATA_DIR / "prepared_data.csv.xz"), index_col=0, keep_default_na=False) with tempfile.TemporaryDirectory(prefix="lookout_typos_fasttext_") as temp_dir: config = {"size": 100, "path": os.path.join(temp_dir, "ft.bin"), "dim": 5} train_fasttext(data, config) model = FastText.load_fasttext_format(config["path"]) self.assertTupleEqual(model.wv["get"].shape, (5,))
def cli_train_fasttext(data_path: str, config: Mapping[str, Any]) -> None: """Entry point for `train_fasttext`.""" train_fasttext( pandas.read_csv(data_path, index_col=0, keep_default_na=False), config)