NoneFast Text с Python: вычислить точность

Question

Я использую fastText с Python, который дает точность и отзыв, но не точность. Как мне получить точность от fastText? Или, как альтернатива, как я могу рассчитать точность с учетом точности и вспомнить?

Answer 1

Я сделал этот код, взяв данные из строки в CSV (начиная с метки) и сравнив с прогнозом, затем он сохраняет в.txt

f = open('accuracy.txt', 'w')
total, correct = 0, 0
for idx, row in X.iteritems():

    #Getting data from a CSV
    line = row.split()
    label = line[0]
    description = " ".join(line[1:])

    #Predicting
    predict = model.predict(description , k=1)

    #Saving accuracy
    total += 1
    if(predict[0][0] == label):
        correct += 1

f.writelines("Accuracy = " + str(correct/total) + '\n')
f.close()