fasttext 训练测试

import fasttext
#训练模型
trainDataFile = '训练文件路径'

classifier = fasttext.train_supervised(
    input = trainDataFile,
    label_prefix = '__label__',
    dim = 256,
    epoch = 50,
    lr = 1,
    lr_update_rate = 50,
    min_count =2,
    loss = 'softmax',
    word_ngrams = 3,
    bucket = 1000000)
classifier.save_model("名称.bin") 
#load训练好的模型
#classifier = fasttext.load_model('名称.model.bin', label_prefix='__label__')

result = classifier.test("测试晚间路径")
print ('测试集上数据量', result[0])
print ('测试集上准确率', result[1])
print ('测试集上召回率', result[2])