我正在尝试使用快速文本 API 构建一个有监督的分类器。我的数据是“output.txt”,有 15000 行、2 列(性别和姓名)和 2 个类 m/f。
__label__F Mary
__label__F Santa
...
__label__M John
代码:
#model = fasttext.load_model('model.bin') # using the default model
classifier = fasttext.supervised('output.txt','model', label_prefix='__label__',thread=8)
labels = classifier.predict('Maria',k=1)
print(labels)
当我用一个名字进行测试时,它会返回
[[],[],[],[]]
我不明白我的错误在哪里。我按照fastext github中的代码行。当相同的代码能够在 Linux 中运行时,在 Windows 中创建的问题是什么?
我正在使用 Windows,python 3.4x。
更新:我已经评论了模型 = fasttext.load .....
以便将学习到的监督分类器模型保存在model.bin中。相同的代码在 Ubuntu 中有效,但在 Windows 中无效。任何帮助表示赞赏。