安装 fasttext python

FastText是Facebook AI研究院所开发的一个基于文本分类和表示学习的快速文本分类库。FastText不仅拥有Word2Vec的学习算法，而且还能够进行文本分类、文本标注和文本表示。

在本文中，我们将讲述如何在Python中安装和使用FastText。

安装

使用pip安装fasttext：

pip install fasttext

使用

通过下面的代码段，我们可以在Python中使用FastText对文本进行分类：

import fasttext

# 创建训练文件
with open('train.txt', 'w') as f:
    f.write('__label__positive The movie was great and entertaining.\n')
    f.write('__label__negative The movie was terrible and boring.\n')

# 训练模型
model = fasttext.train_supervised('train.txt')

# 预测分类
predict_label = model.predict(['The movie was great!'])[0][0]
print(predict_label)

上面这段代码的输出结果应该是__label__positive。

参考

• FastText Github Repository
• Python Package Index Page